將預測結果保存為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'

我正在使用Pandas / 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=',')