在 VScode 中将 tradingview 的 pine 脚本转换为 python

Question

我想将以下 Pine 脚本转换为 python 来计算 vwap1 变量的值而不绘制结果，我只想计算 vwap1 的值：

...
wapScore(pds) =>
    mean = sum(volume*close,pds)/sum(volume,pds)
    vwapsd = sqrt(sma(pow(close-mean, 2), pds) )
    (close-mean)/vwapsd

vwap1 = input(48)
plot(vwapScore(vwap1),title="ZVWAP2-48",color=#35e8ff, linewidth=2,transp=0.75)
...

我尝试了以下方法：

...
def calculate_SMA(ser, days):
    sma = ser.rolling(window=days).mean()
    return sma


def calculate_Zscore(pds, volume, close):
    mean = sum(volume * close, pds) / sum(volume, pds)
    vwapsd = np.sqrt(calculate_SMA(pow(close - mean, 2), pds))
    return (close - mean) / vwapsd

...

我正在使用 calculate_Zscore 函数来计算值并将其添加到 pandas 数据框，但它给了我不同的值而不是交易视图上的值

Answer 1

我只是想发表评论......但我的声誉不允许我:)

当然，pine 脚本 (tradingview) 的 sum 与 python sum 具有不同的类型签名。 特别是第二个参数具有完全不同的含义。 在 pine 脚本中sum(x,y) 告诉您 x 的最后 y 值的滑动总和（y 条后面的 x 总和）。 在python中sum(x,y) sum iterable x 如果y，第二个参数已经传递（可选），这个值被添加到iterable的项目总和中。所以如果你的sum(x) == 4.5 然后@ 987654332@ == 14.5

因此，至少在使用此方法时，您的代码肯定需要更改

希望对你有所帮助