Python - 将ARIMA预测数据保存到文件中

时间:2017-08-26 23:39:04

标签: python

嗨,谢谢你,

所以我正在尝试将我的预测数据写入我使用ARIMA预测的情节中的文件。我怎么能这样做,以便我以后可以访问保存的预测数据?

这是我的代码:

import pandas
from matplotlib import pyplot

series = pandas.read_csv('Quantity.csv',header=0,parse_dates=[0])
series.columns = ['Date','Quantity']
series.set_index(['Date'],inplace=True)

model = ARIMA(series['Quantity'].astype(float), order=(2,0,1))

fittedModel = model.fit(disp=0,method='css')

stepsAhead = 10 
forecastArray = fittedModel.forecast(steps=stepsAhead)
for i in range(stepsAhead):
    series.loc[len(series)] = forecastArray[0][i]

series.plot()

pyplot.show()

如果需要,以下是我用于绘制的数据:

Date    Quantity
2010/01/01  1358
2010/07/02  0
2010/08/03  0
2011/02/04  0
2011/11/05  0
2011/12/06  274
2012/06/07  1074
2012/08/30  2223
2013/04/16  0
2013/03/18  1753
2014/02/22  345
2014/01/27  24
2015/12/15  652
2015/09/28  275
2016/05/04  124
2017/11/07  75
2017/09/22  32
2017/04/04  12

谢谢。

1 个答案:

答案 0 :(得分:0)

series.to_csv(yourPath +'\Forecasts.csv')

我猜这对你很方便。