用于将预测列附加到数据框架的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'], index=qualityTrain.index)...

我得到了我期望的结果。

是否有更好的方式将结果附加到数据框架，其中数据框架可能具有任意索引?

Answer 1

你可以直接做

qualityTrain['Pred1'] = pred1

请注意，我们(统计模型)很快就会有熊猫进熊猫出的预测，所以它有望缓解这些痛点。