将任何 sklearn 结果映射回原始数据帧

mapping back any sklearn result to the original dataframe

我想以原始测试数据为参考,分析我的随机森林结果在excel中的预测值。

当我使用这个时,预测结果出现在一个数组中:

predict = rf.predict(test[columns])

如何使用 pandas 将预测结果映射回原始数据集?

此致, galeej

EdChum 回答了问题。

(编辑)他的回答:

You should be able to just add this back as a new column: test['prediction'] = rf.predict(test[columns])

对于延迟,我深表歉意。

此致, 苏尼尔