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)