Сохранение результатов прогнозирования в 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=',')