用于将预测列附加到数据框的 Pandas 习语

Question

将预测结果附加到进行预测的数据帧的 Pandas 习语是什么。

例如，如果我有类似的东西（其中qualityTrain 是stats models fit 的结果）

qualityTrain = quality_data[some_selection_criterion]
pred1 = QualityLog.predict(qualityTrain)
qualityTrain = pd.concat([qualityTrain, pd.DataFrame(pred1, columns=['Pred1'])], axis=1)

“Pred1”值与qualityTrain 的其余部分未正确对齐。如果我修改最后一行以读取

...pd.DataFrame(pred1, columns=['Pred1'], <b>index=qualityTrain.index</b>)...

我得到了我期望的结果。

有没有更好的方式将结果附加到数据帧可能具有任意索引的数据帧？

Answer 1

你可以这样做

qualityTrain['Pred1'] = pred1

请注意，我们（statsmodels）很快就会有 pandas-in、pandas-out 来进行预测，因此它有望缓解其中的一些痛点。