파이썬(Python) pickle을 이용한 파이썬 객체 저장

---

pickle은 파이썬(Python) 내장 모듈로써 파이썬 객체를 객체 자체로 파일로 저장할 수 있게 해주는 모듈이다. 예를 들면 리스트는 리스트로 딕셔너리는 딕셔너리로 저장된다는 것이다. 따라서 불러올 때에도 객체가 보존되어 바로 사용할 수 있다.

 

pickle이 좋은 이유는 기존 text 파일과 비교해 보면 되는데 예를 들어 5GB짜리 text 파일을 읽어 온 다음 이 중에서 여러 곳에 흩어져 있는 문자열을 가져온다고 해보자. 그러면 전체 text를 파싱하여 필요한 부분을 찾아야 하므로 굉장히 비효율적이다. 하지만 pickle은 여러 곳에 흩어져 있는 문자열을 리스트에 담아서(물론 처음에는 text 파일을 불러와야 한다) 그 리스트 자체를 저장하면 다음부터는 불러올 때 전체 text 파일을 읽어올 필요가 없어져서 굉장히 빠르게 필요한 정보를 불러올 수 있게 된다.

 

이번 포스팅에서는 pickle을 이용하여 파이썬(Python) 객체를 저장하고 불러오는 방법을 알아보려고 한다.

---

   pickle

1) 저장하기 : dump

파이썬 객체 저장은 dump를 이용하면 된다. dump의 첫 번째 인자에 저장하고자할 객체를 넣어주면 된다. 이때 파일명은 '.pickle' 또는 '.pkl'로 끝나야 한다.

 

import pickle

str_list = ['Hello', 'World']
with open('./str_list.pkl', 'wb') as f:
    pickle.dump(str_list, f)

---

2) 불러오기 : load

'.pickle' 또는 '.pkl' 파일은 load를 이용하여 불러올 수 있다.

 

with open('./str_list.pkl', 'rb') as f:
    data = pickle.load(f)
print(data)

 

---