본문 바로가기

연구/pytorch

[파이썬] 피클 pickle 파일 읽기, 쓰기

반응형

딥러닝 공부를 하며 코드들을 다루다 보면 피클 형식으로 저장되어 있는 데이터를 불러오거나 데이터를 피클 형식으로 저장해야 하는 경우가 더러 있다.

 

그렇다면 피클 pickle 이란 무엇일까?

 

텍스트 상태의 데이터가 아닌 파이썬 객체 자체를 파일로 저장하는 것을 말한다.

이 때 원하는 객체 자체를 바이너리로 저장해놓는 것이고 필요할 때 불러오기만 하면 되기 때문에 속도가 빠르다는 장점이 있다.

 

각설하고 피클 사용법은 아래와 같다.

import pickle

my_data = ['a','b','c']
 
### 피클 파일로 저장할 때 ###
with open("data.pickle","wb") as fw:
    pickle.dump(my_list, fw)
 
### 피클 파일 불러오기 ###
with open("data.pickle","rb") as fr:
    data = pickle.load(fr)

print(data)
#['a', 'b', 'c']

 

반응형