SARIMA模型预测python代码
时间: 2023-10-05 11:06:11 浏览: 121
SARIMA-LSTM混合模型预测时间序列(Python)
5星 · 资源好评率100%
SARIMA模型是一种常用的时间序列预测模型,它结合了季节性差分自回归移动平均模型(Seasonal Autoregressive Integrated Moving Average,SARIMA)和自回归移动平均模型(Autoregressive Integrated Moving Average,ARIMA)的特点。下面是一个示例的Python代码,用于使用SARIMA模型进行时间序列预测:
```python
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from statsmodels.tsa.statespace.sarimax import SARIMAX
# 读取时间序列数据
data = pd.read_csv('data.csv', index_col='date', parse_dates=True)
# 拟合SARIMA模型
model = SARIMAX(data, order=(1, 1, 1), seasonal_order=(1, 1, 1, 12))
result = model.fit()
# 进行预测
start_date = '2022-01-01'
end_date = '2022-12-31'
prediction = result.predict(start=start_date, end=end_date)
# 绘制原始数据和预测结果的图像
plt.plot(data, label='Actual')
plt.plot(prediction, label='Prediction')
plt.legend()
plt.show()
```
在这个示例代码中,我们假设时间序列数据保存在名为"data.csv"的文件中,其中包含一个"date"列和一个"value"列,表示日期和对应的值。首先,我们使用pandas库读取数据。然后,我们创建了一个SARIMA模型并使用拟合方法对模型进行拟合。在拟合过程中,我们指定了模型的阶数(order)和季节性阶数(seasonal_order)。最后,我们使用训练好的模型进行预测,并将结果与原始数据一起绘制在图像上。
请注意,这只是一个示例代码,实际使用SARIMA模型时,您可能需要根据数据的特点进行适当的参数调整和模型选择。
阅读全文