如何在python中使用移动平均?

2024-03-29 00:33:52 发布

您现在位置:Python中文网/ 问答频道 /正文

我想用简单的移动平均法做一个时间序列预测。我使用以下代码:

from statsmodels.tsa.arima_model import ARMA
import statistics
data=[x + random() for x in range(1,100)]
model=ARMA(data,order=(0,1))
model_fit=model.fit(disp=False)
y_hat=model_fit.predict(len(data),len(data))

我不知道如何预测下10个值,因为你只给了我1个值。另外,yèhat与数据的平均值不匹配,因为我使用的是MA中的顺序1。有人能帮我吗?你知道吗


Tags: 代码fromimportdatamodellenhat时间
2条回答
y_hat=model_fit.predict(len(data),len(data))

你的开始值是len(data),结束值也是一样的,所以它给你一个预测值。你知道吗

我在我的项目中使用了forecast()。下面是我的代码片段:

from statsmodels.tsa.stattools import acf

# Create Training and Test
train = df[:3000]
test = df[3000:]

# Build Model
# model = ARIMA(train, order=(3,2,1))  
model = ARIMA(train, order=(1, 1, 1))  
fitted = model.fit(disp=-1)  

# Forecast
fc, se, conf = fitted.forecast(len(test), alpha=0.05)  # 95% conf

# Make as pandas series
fc_series = pd.Series(fc, index=test.index)
lower_series = pd.Series(conf[:, 0], index=test.index)
upper_series = pd.Series(conf[:, 1], index=test.index)

# Plot
plt.figure(figsize=(12,5), dpi=100)
plt.plot(train, label='training')
plt.plot(test, label='actual')
plt.plot(fc_series, label='forecast')
plt.fill_between(lower_series.index, lower_series, upper_series, 
                 color='k', alpha=.15)
plt.title('Forecast vs Actuals')
plt.legend(loc='upper left', fontsize=8)
plt.show()

enter image description here

预测值在我的图表中是恒定的,因为我的数据有季节性成分。你知道吗

我想你只需要像这样给出起始值和结束值

 model_fit.predict(0,10)

相关问题 更多 >