arima模型python代码
时间: 2023-09-08 09:14:31 浏览: 23
以下是一个简单的ARIMA模型的Python代码示例:
```python
# 导入必要的库
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from statsmodels.tsa.arima_model import ARIMA
# 读取数据
df = pd.read_csv('data.csv', parse_dates=['date'], index_col='date')
# 拆分训练集和测试集
train = df.loc['2010-12-01':'2019-12-31']
test = df.loc['2020-01-01':]
# 拟合ARIMA模型
model = ARIMA(train, order=(1, 1, 1))
results = model.fit()
# 预测
start = len(train)
end = len(train) + len(test) - 1
predictions = results.predict(start=start, end=end, typ='levels').rename('ARIMA Predictions')
# 绘制预测结果和实际数据
plt.figure(figsize=(10,5))
plt.plot(test)
plt.plot(predictions, color='red')
plt.title('ARIMA Model')
plt.legend(['Actual', 'Predicted'])
plt.show()
```
注意,这只是一个基本的ARIMA模型示例,需要根据具体情况进行调整和优化。