建立arima模型的代码

ARIMA模型是一种基于时间序列的预测模型，可以用来预测未来的时间序列值。以下是建立ARIMA模型的代码示例：

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from statsmodels.tsa.arima_model import ARIMA

# 读入数据
data = pd.read_csv('data.csv', index_col='date', parse_dates=['date'])

# 拆分训练集和测试集
train_data = data.iloc[:len(data)-12]
test_data = data.iloc[len(data)-12:]

# 建立ARIMA模型
model = ARIMA(train_data, order=(1, 1, 1))
model_fit = model.fit(disp=0)

# 预测未来12个月的数据
forecast = model_fit.forecast(steps=12)[0]

# 绘制预测结果
plt.plot(test_data, label='Actual')
plt.plot(forecast, label='Predicted')
plt.legend()
plt.show()

这段代码假设数据文件名为data.csv，其中包含日期和时间序列值。首先读入数据，然后将数据拆分为训练集和测试集。接着，我们使用ARIMA模型来拟合训练集中的数据，并设置ARIMA模型的参数为(1, 1, 1)。最后，我们使用训练好的模型来预测未来12个月的数据，并将预测结果与测试集中的实际值进行比较，绘制出预测结果和实际值的曲线。