將預測結果保存爲CSV

Question

我將sklearn迴歸模型的結果存儲到varibla預測中。

prediction = regressor.predict(data[['X']]) 
print(prediction) 

預測輸出的值看起來像這樣

[ 266.77832991 201.06347505 446.00066136 499.76736079 295.15519906 
    214.50514991 422.1043505 531.13126879 287.68760191 201.06347505 
    402.68859792 478.85808879 286.19408248 192.10235848] 

然後我試圖使用to_csv函數將結果保存到本地CSV文件：

prediction.to_csv('C:/localpath/test.csv') 

但我得到的錯誤是：

AttributeError: 'numpy.ndarray' object has no attribute 'to_csv' 

我正在使用熊貓/ Numpy/SKlearn。任何想法的基本修復？

Answer 1

您可以使用熊貓。據說，numpy數組沒有to_csv函數。

import numpy as np 
import pandas as pd 
prediction = pd.DataFrame(predictions, columns=['predictions']).to_csv('prediction.csv') 

如果您希望您的值爲行或列，請添加「.T」。

Answer 2

可以使用numpy.savetxt功能：

numpy.savetxt('C:/localpath/test.csv',prediction, ,delimiter=',') 

，並加載一個CSV文件，您可以使用numpy.genfromtxt功能：

numpy.genfromtxt('C:/localpath/test.csv', delimiter=',')