python - 将预测结果保存为CSV

我将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。对基本修复有什么想法吗？

最佳答案

你可以用熊猫。
如前所述，numpy数组没有to-csv函数。

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

如果您希望在类似行或列中输入值，请添加“.t”。