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

Question

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

预测的结果进入数组，因为我使用此：

predict = rf.predict(test[columns])

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

问候，galeej

Answer 1

EdChum回答了这个问题。

（编辑）他的答案：

您应该可以将其重新添加为新列：test ['prediction'] = rf.predict（test [columns]）

对于延迟，我深表歉意。

问候，苏尼尔