有时候做完一个案例的分析,模型做好了,最后想把预测的结果保存为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})