python-逻辑回归，将预测概率和预测保存到csv

Question

我是数据分析和 python 的新手。我一直在关注这个链接中关于泰坦尼克号幸存者的逻辑回归：

http://hamelg.blogspot.ca/2015/11/python-for-data-analysis-part-28.html

不过，我正在使用我自己的非泰坦尼克数据集。我在要将结果导出到 csv 文件的示例的末尾。我做了一个小修改，但这让我陷入了困境。除了预测之外，我还明确生成了预测概率，我还想将其导出到 csv 文件中。

test_probs=log_model.predict_proba(X=test_features)
print(test_probs)


# Create a submission for Kaggle
submission = pd.DataFrame({"AccountNumber":titanic_test["AccountNumber"],
"PolarPredict":test_preds,**"probabilities":test_probs** })

这是我得到的消息：异常：数据必须是一维的

这是教程的原始代码： # 进行测试集预测 test_preds = log_model.predict(X=test_features)

# Create a submission for Kaggle
submission = pd.DataFrame({"PassengerId":titanic_test["PassengerId"],
                       "Survived":test_preds})

# Save submission to CSV
submission.to_csv("tutorial_logreg_submission.csv", 
              index=False) 

如何将预测、概率和“ID”导出到 csv 文件中？

Answer 1

您可以尝试将它们转换为列表

submission = pd.DataFrame({
     "AccountNumber":list(titanic_test["AccountNumber"]),
     "PolarPredict":list(test_preds),
     "probabilities":list(test_probs)
     })

submission.to_csv("tutorial_logreg_submission.csv", 
          index=False)