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

Question

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

下面是我开发的代码，代码基本就是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 函数在内部如何工作，以便我可以实现它的克隆？

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

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

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

How TradingView Pine Script RMA function works internally?

moving-average

pine-script