将随机森林预测作为列添加到测试文件中

将随机森林预测作为列添加到测试文件中

以来

rfc_model_3 = RandomForestClassifier(n_estimators=200)rfc_model_3.predict(X_test)

返回

y : array of shape = [n_samples]

（请参阅docs），您应该能够直接将模型输出添加到其中，

X_test

而无需创建中间体

Dataframe

：

X_test['survived'] = rfc_model_3.predict(X_test)

如果您仍然想要中间结果，则@EdChum在评论中的建议会很好地工作。