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)

＆＃39; Pred1＆＃39;值未与qualityTrain的其余值正确对齐。如果我修改最后一行以便读取

...pd.DataFrame(pred1, columns=['Pred1'], index=qualityTrain.index)...

我得到了我期望的结果。

将结果附加到数据框可能有任意索引的数据框中是否有更好的习惯用法？

Answer 1

你可以做到

qualityTrain['Pred1'] = pred1

请注意，我们（statsmodels）会很快就会有pandas-in，pandas-out用于预测，所以它有望减轻这些痛点。