TradingView Pine Script RMA 功能如何在内部工作？

Question

我正在尝试重新实现该rma函数，TradingView pinescript但无法使其输出与原始函数相同的结果。

下面是我开发的代码，代码基本就是ema函数，但是做图的时候和rma函数plot结果差别很大：

//@version=3
study(title = "test", overlay=true)

rolling_moving_average(data, length) =>
    alpha = 2 / (length + 1)
    sum = 0.0
    for index = length to 0
        if sum == 0.0
            sum := data[index]
        else
            sum := alpha * data[index] + (1 - alpha) * sum

atr2 = rolling_moving_average(close, 5)
plot(atr2, title="EMAUP2", color=blue)

atr = rma(close, 5)
plot(atr, title="EMAUP", color=red)

所以我的问题是该rma函数如何在内部工作，以便我可以实现它的克隆？

附注。这是文档的链接https://www.tradingview.com/study-script-reference/#fun_rma它确实显示了可能的实现，但在运行时不起作用。

Answer 1

下面是正确的实现：

plot(rma(close, 15))

// same on pine, but much less efficient
pine_rma(x, y) =>
	alpha = 1/y
    sum = 0.0
    sum := alpha * x + (1 - alpha) * nz(sum[1])
plot(pine_rma(close, 15))

TradingView 上的代码有错误，alpha 应该是 1/y 而不是 y。这个维基百科页面有 RMA维基百科的正确公式 - 移动平均线