moving-average - 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 公式
Wikipedia - moving averages