时间序列分析方法之 -- 自回归移动平均模型（Autoregressive Moving Average, ARMA）原理及Python代码示例

原理

自回归移动平均模型（Autoregressive Moving Average, ARMA）是一种结合了自回归（AR）模型和移动平均（MA）模型的时间序列分析模型。它适用于描述和预测平稳时间序列数据。ARMA 模型将时间序列的当前值表示为其过去值和过去预测误差的线性组合。

ARMA(p, q) 模型的数学表达式如下：

其中：

表示时间 t 的实际值。
c 是常数项（截距）。
是自回归系数。
是移动平均系数。
是白噪声误差项，假设为均值为零且方差为常数的独立同分布随机变量。

ARMA 模型中包含两个部分：

自回归（AR）部分：表示当前值与过去值之间的关系。
移动平均（MA）部分：表示当前值与过去误差之间的关系。

适用情况

ARMA 模型适用于以下情况：

平稳时间序列：ARMA 模型适用于平稳时间序列，即时间序列的均值、方差和自协方差在时间上不随时间变化。
无明显季节性：ARMA 模型不适用于具有明显季节性模式的时间序列。对于具有季节性特征的时间序列，可以考虑使用季节性自回归移动平均模型（SARMA）或其他季节性时间序列模型。
短期预测：ARMA 模型通常用于短期预测，因为它依赖于时间序列的过去值和误差项。

Python 示例代码

以下是使用 Python 实现自回归移动平均模型的示例代码，利用 statsmodels 库进行建模和预测：

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from statsmodels.tsa.arima.model import ARIMA
from statsmodels.tsa.stattools import adfuller

# 生成样本数据
np.random.seed(0)
data = np.random.randn(100).cumsum()

# 创建数据序列
data_series = pd.Series(data)

# 检查数据平稳性
result = adfuller(data_series)
print('ADF Statistic:', result[0])
print('p-value:', result[1])

# 差分处理使数据平稳
if result[1] > 0.05:
    data_series = data_series.diff().dropna()

# 应用ARMA模型
p = 2  # 自回归阶数
q = 2  # 移动平均阶数
model = ARIMA(data_series, order=(p, 0, q)).fit()
print(model.summary())

# 进行预测
pred_start = len(data_series)
pred_end = len(data_series) + 10
predictions = model.predict(start=pred_start, end=pred_end, dynamic=False)

# 可视化结果
plt.figure(figsize=(12, 6))
plt.plot(data_series, label='Original Data')
plt.plot(range(pred_start, pred_end + 1), predictions, label='Predictions', color='red')
plt.legend()
plt.title('Autoregressive Moving Average Model (ARMA)')
plt.show()

在上述代码中：

我们首先生成了一些样本数据，并将其创建为一个 Pandas 系列对象。
使用 adfuller 函数检查数据的平稳性。如果数据不是平稳的，可以通过差分处理使其平稳。
使用 ARIMA 类拟合 ARMA 模型，并打印模型摘要。注意，这里我们指定了 ARMA(p, q) 模型的阶数 p 和 q。
使用拟合好的模型进行预测，并将预测结果与原始数据进行可视化。

结论

自回归移动平均模型（ARMA）是一种有效的时间序列分析和预测方法，通过结合自回归和移动平均两种模型的优势，可以更好地捕捉时间序列的动态特征。ARMA 模型特别适用于平稳且无明显季节性模式的时间序列数据。通过选择合适的模型阶数，可以显著提高预测的准确性。在实践中，结合数据的具体特征和目标需求，调整模型参数以获得更好的预测效果。

wodertianna

时间序列分析方法之 -- 自回归移动平均模型（Autoregressive Moving Average, ARMA）原理及Python代码示例

结论