有時候做完一個案例的分析,模型做好了,最後想把預測的結果儲存為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})

.