python 怎样把一个数组类型数据保存为csv文件

有时候做完一个案例的分析,模型做好了,最后想把预测的结果保存为CSV文件。这个预测的结果一般是n行1列的一个数列,我们最后就是想把预测结果保存下来。

submission = clf.predict(credit_test)#submission是我的预测结果,是一个5000*1的列向量

submission = pd.DataFrame({'RISK':submission})

#必须把这个submission用pd.DataFrame读出来

不然会报错 AttributeError: 'numpy.ndarray' object has no attribute 'to_csv'

submission.to_csv('submission.csv',index=False)


以后遇到“AttributeError: 'numpy.ndarray' object has no attribute 'to_csv'”就可以用上面的方法解决

也可以用这个submission=pd.Series(submission)替换submission = pd.DataFrame({'RISK':submission})

你可能感兴趣的:(python解决的问题)