解决报错:The indices for endog and exog are not aligned
在进行ARIMAX模型训练的时候,我让ChatGPT帮助我写了读取数据和训练模型的全过程,但在运行的时候出现了报错:The indices for endog and exog are not aligned。
具体代码如下:
import numpy as np
import pandas as pd
import statsmodels.api as sm
# 从Excel文件中读取数据,假设数据包含一列名为'年份'的时间序列数据和列A、B、C、D
df = pd.read_excel('/path/to/your/data.xlsx') # 替换成你的Excel文件路径
# 提取时间序列数据
ts_data = df.set_index('年份')['A'] # 替换成你的时间序列列名
# 提取外部变量
exog_data = df('年份')[['B', 'C', 'D']]
# 构建 ARIMAX 模型
# 假设使用 ARIMA(1, 0, 0) 模型,你可以根据实际情况调整阶数
arimax_model = sm.tsa.ARIMA(ts_data, order=(1, 0, 0), exog=exog_data)
# 拟合 ARIMAX 模型
arimax_result = arimax_model.fit()
# 输入新的外部变量值进行预测
new_B = 1200000 # 替换成你的新的 B 值
new_C = 650000 # 替换成你的新的 C 值
new_D = 38000 # 替换成你的新的 D 值
# 构建新的外部变量 DataFrame,并确保索引对齐
new_exog_data = pd.DataFrame({'B': [new_B], 'C': [new_C], 'D': [new_D]}, index=[ts_data.index[-1]])
# 使用 predict 函数进行预测
predicted_A = arimax_result.predict(start=len(ts_data), end=len(ts_data), exog=new_exog_data)
# 输出预测值
print("预测的 A 值为:", predicted_A)
报错如下图所示:
查阅相关资料后,发现一些文章就其他模型遇到的相关问题进行了解答,参考如下:
https://blog.csdn.net/qq_18999357/article/details/86595741
https://segmentfault.com/q/1010000042827441?utm_source=sf-similar-question
https://qa.1r1g.com/sf/ask/3554388981/
根据以上文章,我对自己的数据进行了自查,发现数据的矩阵大小没有问题,问题出在了索引上。我的因变量的索引被赋值为“年份”,但是自变量没有赋值,所以默认从0开始顺序编号。所以因变量和自变量的索引不同,只需要更改自变量的索引就可以结局问题。
将原来的代码:
exog_data = df('年份')[['B', 'C', 'D']]
替换为:
exog_data = df.set_index('年份')[['B', 'C', 'D']]
exog_data.set_index(ts_data.index, inplace=True)
即可解决。