如何在 python 中绘制移动平均线？

Question

如何制作 y_pred_org 的移动平均线并绘制它？我已经按照以下方式尝试过，但收到错误 AttributeError: 'list' object has no attribute 'rolling' 我确信这是小事，但我是新手。

# Visualize the prediction with rolling average
from matplotlib import pyplot as plt
plt.figure()
['y_pred_org'].rolling(window=50).mean().plot()
plt.plot(y_test_t_org)
plt.title('Prediction vs Real Stock Price')
plt.ylabel('Price')
plt.xlabel('Days')
plt.legend(['Prediction', 'Real'], loc='upper left')
#plt.show()

我应该如何调整它才能正常工作？以下代码将正常运行。如果需要相移，也可以添加shift（任何你需要的shift）

# Visualize the prediction with rolling average
from matplotlib import pyplot as plt
plt.figure()
df = DataFrame(data = y_pred_org)
df.rolling(30, center=True).mean().plot()
plt.plot(y_test_t_org)
plt.title('Prediction vs Real Stock Price')
plt.ylabel('Price')
plt.xlabel('Days')
plt.legend(['Prediction', 'Real'], loc='upper left')
#plt.show()

Answer 1

如果您的数据在 pandas DataFrame 中，您可以使用内置滚动平均值。

df['Y_Predict'] = df.iloc[:,col].rolling(window=5).mean()

如何在 python 中绘制移动平均线？

How do I plot a moving average in python?

python

machine-learning

matplotlib

moving-average