Zapisywanie wyników prognozowania na CSV

Question

Przechowuję wyniki z modelu regresji sklearn do prognozy varibla.

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

Wartości wyjścia predykcji wygląda to

[ 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] 

ja potem próbuje użyć funkcji to_csv aby zapisać wyniki do lokalnego pliku CSV:

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

But błąd, który otrzymuję, to:

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

Używam Pandas/Numpy/SKlearn. Masz pomysł na podstawową poprawkę?

Answer 1

Możesz użyć pand. Jak już wspomniano, numpy tablice nie mają funkcji to_csv.

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

dodać ".T", jeśli chcesz albo wartości w linii lub w kolumnie.

Answer 2

Można użyć numpy.savetxt funkcję:

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

i wczytać plik CSV można użyć numpy.genfromtxt funkcję:

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