python实现 ARIMA 模型 及逐行解释

2744557306

ARIMA（Autoregressive Integrated Moving Average）模型是一种常用的时间序列分析方法，用于对时间序列数据建模和预测。它通过结合自回归（AR）和滑动平均（MA）模型的特性，并对序列进行差分（Integration，即I）来建立模型。
ARIMA模型的三个主要参数是p、d和q，分别对应于自回归、差分和滑动平均的阶数。
% s) F/ J" w9 J8 e: w" P$ o
1.自回归（AR）：自回归部分使用先前时间点的观测值来预测当前值。p参数表示自回归的阶数，即使用多少个先前时间点的值作为预测输入。
  F! ?- d6 K- b2.差分（I）：差分是对时间序列进行一阶或多阶的差分操作，可以消除序列的非平稳性。d参数表示差分的阶数，默认为1阶差分。
3.滑动平均（MA）：滑动平均部分使用先前的误差值来预测当前值。q参数表示滑动平均的阶数，即使用多少个先前的误差值作为预测输入。

ARIMA模型的一般表示形式为ARIMA(p, d, q)，其中p、d和q是非负整数。它可以很好地处理具有线性趋势和季节性的时间序列数据。


4.确定时间序列数据的平稳性，如平稳性检验、观察序列的趋势和季节性等。
5.如果时间序列不平稳，进行差分操作以实现平稳性。
' k& E/ \1 }( O$ K' C' f3 L* f6.通过观察ACF和PACF图来确定p和q的合适取值范围。
7.根据AIC等准则，以不同的p、d、q值建立多个ARIMA模型。
8.对每个模型进行参数估计和模型拟合。
9.使用拟合的模型进行预测，并对模型的拟合效果进行评估。

5 c2 }1 m. w7 y; O8 |ARIMA模型是时间序列分析领域中常用的模型之一，它可以用于预测未来趋势、季节性和周期性等时间序列数据的变化。在实际应用中，ARIMA模型可以被用于经济预测、股票市场分析、天气预测等各种领域。
( C- v7 X  O2 a. I4 a1 F希望这个介绍对你有帮助。如果你还有任何问题，请随时提问。
import numpy as np
import pandas as pd
from statsmodels.tsa.arima_model import ARIMA
) ~+ ?/ ?9 [  T' T6 ?from statsmodels.graphics.tsaplots import plot_acf, plot_pacf
import matplotlib.pyplot as plt

这几行代码导入了所需的库，包括NumPy（用于数值计算）、Pandas（用于数据处理和分析）、statsmodels库中的ARIMA模型类和绘制时间序列图的函数，以及matplotlib用于绘图。
) i$ B" `5 q! q. g. B. h, kdf = pd.DataFrame({
+ R  W1 u1 H+ J. [6 J    'year': [i for i in range(1980, 2011)],
    'val': [0.82989428, 0.85951092, 0.87668916, 0.86670716, 0.932052,
8 P5 X3 H: l7 v4 G  h            1.04826364, 1.3111932, 1.63756228, 2.0641074, 1.91268276,
- U$ T0 u. f* h$ M9 U            2.03544572, 2.17721128, 2.38968344, 2.75059208, 3.0906664,
' o5 J6 U/ n5 Q; v; X0 n$ A            3.42664028, 3.83064908, 3.97190864, 3.83160036, 4.143101,
' p7 o+ p0 \4 D0 z% D6 p! H9 i            4.566551, 4.47541, 4.462796, 4.384829, 4.796861,
# ^2 c; E7 Y+ U$ D% u1 s& R6 _: C            5.046211, 5.098759, 5.196519, 5.166843, 5.174744,
" [3 u2 L5 A8 j/ ~5 a            5.440894],
})

这段代码定义了一个名为df的Pandas DataFrame，其中包含了两列数据：'year'和'val'。'year'列包含了从1980年到2010年的年份数据，'val'列包含了与每个年份对应的值。
) U  k$ e4 [# b$ h" odf['val_diff1'] = df['val'].diff()
plt.plot(df['year'], df['val'], label='origin')
" b6 g3 n) F/ S* j: ]+ U/ Lplt.plot(df['year'], df['val_diff1'], label='diff1')
$ z3 q, O% k6 a" j3 ~plt.legend()

这段代码将计算'val'列的一阶差分，并将结果存储在一个新的'valdiff1'列中。然后使用matplotlib绘制了两条线：一条是原始'val'列的线，另一条是差分后的'valdiff1'列的线。plt.legend()函数用于显示图例。
: p' @0 z0 w! V9 y9 F# I2 S" \plot_acf(df['val_diff1'][1:])

2 T9 G8 d) n, R# o) X9 q这行代码使用plot_acf函数绘制了一阶差分后的序列（'val_diff1'列）的自相关函数（ACF）图。ACF图用于展示序列在不同滞后阶数的相关性。
' H! s( ~" R1 P5 xplot_pacf(df['val_diff1'][1:], lags=14)

8 P7 y! u$ R& z8 V# o这行代码使用plot_pacf函数绘制了一阶差分后的序列（'val_diff1'列）的偏自相关函数（PACF）图。PACF图用于展示序列在不同滞后阶数的偏相关性，同时指定了lags=14参数，表示只展示14个滞后阶数的PACF值。
( s) k7 W2 r& E* K. ystr_list = []
) ~" ^& }1 c# M% |; }! w' Ifor p in range(1, 4):
    for q in range(0, 4):
' f1 ]# {3 Y; k% p1 i        model = ARIMA(df['val'], order=(p, 1, q))
9 g* p. E) Z% i1 A3 V. [        res = model.fit(disp=0)
        str_list.append('p = {}, q = {}, aic = {}'.format(p, q, res.aic))
2 M, ]. n- r6 ^7 y* K. tfor each in str_list:
  q6 e; F5 J( D6 Z    print(each)

这段代码使用嵌套的for循环，遍历p和q的取值范围，分别为1到3和0到3。在每次循环中，创建了一个ARIMA模型对象，并将p、1（表示一阶差分）、q作为参数传递给order参数。然后使用拟合方法（fit）将模型拟合到'val'列的数据中，得到拟合后的模型对象（res）。同时，计算并记录了模型的AIC值，并将其添加到字符串列表str_list中。最后，使用循环打印出每个p和q值对应的模型的AIC值。
) G+ F/ m& U- o, _! w; t+ ^* \model = ARIMA(df['val'], order=(2, 1, 0))
) Q% T. W0 l. i! G3 l3 @. C- x( xres = model.fit(disp=0)
5 q, Y3 `+ J1 i, Kres.summary()

这部分代码创建了一个ARIMA模型对象，使用p=2、d=1（一阶差分）、q=0来进行参数配置。然后利用拟合方法将模型拟合到'val'列的数据中，得到拟合后的模型对象（res）。接下来，调用summary()方法打印出拟合后的模型的详细摘要信息，包括模型系数、标准误差、p值等。
+ X' I) [7 z1 Qres.plot_predict(end=40)

这行代码使用拟合后的模型（res）的plot_predict方法生成了一个预测图表。指定了end=40参数，表示要预测40个时间点。这个图表显示了原始数据和模型对未来值的预测。
希望这些解释可以帮助你理解代码的每一行。如果还有进一步的问题，请随时提问。


) f8 u  h' B; h! C9 T9 U