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

Question

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

当我使用这个时，预测的结果出现在一个数组中：

predict = rf.predict(test[columns])

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

问候， 加利杰

Answer 1

EdChum 回答了这个问题。

（编辑）他的回答：

您应该可以将其添加回新列： test['prediction'] = rf.predict(test[columns])

我对延误表示歉意。

问候， 苏尼尔