python - 将数组附加到数据框(python)

Question

因此，我在一个小型销售数据集上运行了一个时间序列模型，并预测了接下来 12 个时期的销售情况。使用以下代码:

 mod1=ARIMA(df1, order=(2,1,1)).fit(disp=0,transparams=True)
    y_future=mod1.forecast(steps=12)[0]

其中 df1 包含以月份为索引的销售值。现在我按以下方式存储预测值:

pred.append(y_future)

现在，我需要将预测值附加到原始数据集 df1，最好使用相同的索引。我正在尝试使用以下代码:

df1.append(pred, ignore_index=False)

但我收到以下错误:

TypeError: cannot concatenate a non-NDFrame object

我试过将 pred 变量转换为列表，然后追加，但无济于事。任何帮助将不胜感激。谢谢。

Answer 1

一种解决方案是使用 df.loc

将新数组附加到您的 dataFrame 的最后位置

df.loc[len(df)] = your_array

但这不是有效的，因为如果您想多次执行此操作，则必须为每个新追加获取 DataFrame 的长度。

更好的解决方案是创建一个包含您需要附加的值的字典，并将其附加到数据帧。

df = df.append(dict(zip(df.columns, your_array)), ignore_index=True)