Python 时间序列预测:使用 ARIMA 模型预测未来 24 个月数据
实现时间序列预测可以使用 Python 中的多种库和算法,比如 ARIMA、SARIMA、LSTM 等。下面是一个使用 ARIMA 模型进行时间序列预测的示例代码:\n\npython\nimport pandas as pd\nimport matplotlib.pyplot as plt\nfrom statsmodels.tsa.arima.model import ARIMA\n\n# 读取数据\ndata = pd.read_csv('data.csv', parse_dates=['date'], index_col='date')\n\n# 按照月份进行重采样\ndata_monthly = data.resample('M').sum()\n\n# 拆分训练集和测试集\ntrain = data_monthly[:-24]\ntest = data_monthly[-24:]\n\n# 训练 ARIMA 模型\nmodel = ARIMA(train, order=(1, 1, 1))\nmodel_fit = model.fit()\n\n# 预测未来 24 个月的数据\nforecast = model_fit.forecast(steps=24)[0]\n\n# 可视化结果\nplt.plot(train, label='Train')\nplt.plot(test, label='Test')\nplt.plot(test.index, forecast, label='Forecast')\nplt.legend()\nplt.show()\n\n\n在代码中,首先读取时间序列数据,并按照月份进行重采样。然后将数据分为训练集和测试集,其中训练集包括过去的数据,测试集包括未来的数据。接下来,使用 ARIMA 模型拟合训练集的数据,并使用该模型进行未来 24 个月的预测。最后,将训练集、测试集和预测结果进行可视化展示。\n\n请确保将data.csv替换为实际的数据文件名,并根据需要调整 ARIMA 模型的参数。
原文地址: https://www.cveoy.top/t/topic/qBQ3 著作权归作者所有。请勿转载和采集!