[PYTHON] 예측 결과를 CSV로 저장

예측 결과를 CSV로 저장

저는 sklearn 회귀 모델의 결과를 다양한 예측에 저장하고 있습니다.

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을 사용하고 있습니다. 기본 수정에 대한 아이디어가 있습니까?

해결법

1. ==============================

   1.팬더를 사용할 수 있습니다. 그것이 말했듯이, numpy 배열에는 to_csv 함수가 없습니다.

   팬더를 사용할 수 있습니다. 그것이 말했듯이, numpy 배열에는 to_csv 함수가 없습니다.

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

   값을 줄 또는 세로로 표시하려면 ".T"를 추가하십시오.

2. ==============================

   2.numpy.savetxt 함수를 사용할 수 있습니다.

   numpy.savetxt 함수를 사용할 수 있습니다.

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

   numpy.genfromtxt 함수를 사용하여 CSV 파일을로드 할 수 있습니다.

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

from https://stackoverflow.com/questions/34864695/saving-prediction-results-to-csv by cc-by-sa and MIT license