Today I leaned..

[Today I learned] 2023.12.14.목: 데이터프레임 파일로 저장하기

Jenny Noh 2023. 12. 14. 18:10

📌 데이터 프레임을 파일로 저장하기

1. csv 파일로 저장하기

데이터 프레임을 csv 파일로 저장하려면 to_csv() 함수를 쓰면된다.

csv 파일로 저장할 때 단점은, 데이터 타입이 날아가기 때문에 다시 불러와서 사용하기 위해서는 추천하지는 않는 방법이다.

 

cf. 파일 open, write, close 함수로 데이터를 csv 파일로 저장하기 위해서는 문자열 형태의 데이터를 입력해야한다. 

(아니면 아래처럼 오류남)

 

2. Pickle 파일로 저장하기

피클은 데이터프레임 뿐만 아니라 파이썬의 모든 객체를 파일로 저장할 수 있는 방법이다. 보통은 sklearn으로 머신러닝 모델을 학습시키고 저장할 때 사용한다고 한다. 

객체를 다시 불러와서 작업하기엔 가장 적합한 저장 방법인 것 같다.

 

피클로 데이터프레임을 저장하려면 to_pickle(), read_pickle()함수를 사용한다.