gpt4 book ai didi

python - 根据 Pandas ewm() 计算指数移动平均线

转载 作者:塔克拉玛干 更新时间:2023-11-03 04:54:18 25 4
gpt4 key购买 nike

我正在尝试构建一个指数移动平均算法,它产生与 Pandas 相同的输出 ewm()功能。特别是,我正在尝试实现这种方法:

enter image description here

下面的代码可以正常工作,直到移动平均窗口开始超出初始数据集,此时我开始得到与 Pandas 计算不同的结果。

我已经看了好几个小时了,现在很困惑。谁能指出我是如何错误地执行上述公式的?

import numpy as np
import pandas as pd

class MovingAverages(object):
def __init__(self, **kwargs):
self.measures = []
self.lookback_period = 5
ema_multiplier = 2 / (self.lookback_period + 1)
self.lookback_alphas = []
for i in range(1,self.lookback_period+1):
self.lookback_alphas.append((1 - ema_multiplier ) ** i)

def insert_bar(self):
self.measures.insert(0, 0)

def on_calculate(self, c):
index = min(len(c), self.lookback_period+1)
y = c[0]
for i in range(1,index):
y += self.lookback_alphas[i-1] * c[i]
y /= 1 + sum(self.lookback_alphas[0:index-1])
self.measures[0] = y

if __name__ == "__main__":
data = [5.00,7.00,4.00,3.00,4.00,
5.00,6.00,7.00,9.00,13.00,
15.00,14.00,14.00,12.00,
11.00,10.00,9.00,8.00,
8.00,8.00,10.00,11.00,
13.00,16.00,18.00,20.00]

# Manually calculate exponential MA and write into list
ma_online = MovingAverages()
series = []
for d in data:
series.insert(0, d)
ma_online.insert_bar()
ma_online.on_calculate(series)

# Calculate a reference exponential MA using Pandas
df = pd.DataFrame({'close': data})
ma_pd = list(df.close.ewm(span=ma_online.lookback_period, adjust=True, ignore_na=True).mean())

# Compare the two lists
ma_online.measures.reverse()
for i in range(len(data)):
assert round(ma_pd[i], 2) == round(ma_online.measures[i], 2)

最佳答案

我不熟悉 Python 或 Panda,但根据我对文档的理解,当你在 df.close.ewm(span=ma_online,...) 中使用 span,你只是在公式中定义衰减(系数),而不是滑动窗口的大小。默认情况下,似乎没有固定的侧窗。

来自 another documentation of DataFrame.ewm :

span : float, optional
Specify decay in terms of span, α=2/(span+1), for span≥1

和:

See also:
rolling: Provides rolling window calculations

rolling 方法可能就是您所需要的。

关于python - 根据 Pandas ewm() 计算指数移动平均线,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50457141/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com