Salvataggio dei risultati di previsione in CSV

Question

Sto memorizzando i risultati da un modello di regressione sklearn alla previsione varibla.

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

I valori della potenza previsione sembra questo

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

Sto quindi cercando di utilizzare la funzione to_csv per salvare i risultati in un file CSV locale:

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

Ma l'errore che torno è:

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

Sto usando Pandas/Numpy/SKlearn. Qualche idea sulla correzione di base?

Answer 1

È possibile utilizzare i panda. Come si dice, gli array numpy non hanno una funzione to_csv.

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

aggiungi ". T" se vuoi i tuoi valori in linea o come colonna.

Answer 2

È possibile utilizzare la funzione numpy.savetxt:

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

e per caricare un file CSV è possibile utilizzare la funzione numpy.genfromtxt:

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