从 Pine 脚本到 Python 的递归函数

如何解决从 Pine 脚本到 Python 的递归函数

我正在尝试将以下 Pine 函数（alpha = 1 / 长度的指数加权移动平均线）转换为 Python 以应用于数据帧 df 的 Pandas 列（下面 Pine 代码中的 src）。 “sma”指的是简单移动平均线，sma[1]指的是前一点的sma：

pine_rma(src,length) =>
alpha = 1/length
sum = 0.0
sum := na(sum[1]) ? sma(src,length) : alpha * src + (1 - alpha) * nz(sum[1])

我已将以下 Python 代码放在一起，但启动递归的最佳方式是什么（第 r=[0] 行）？

def rolling_apply(group,length):
    alpha = 1/length
    r = [0]
    for x in group.x[:-1]:
        r.append(alpha * r + (1-alpha)*r[-1])
    group.r = r
    return group
    
df.r = df.groupby(df.index).apply(rolling_apply).r