[Python] 딕셔너리,리스트 저장하기(pickle)

코딩을 해서 어떤 데이터를 리스트나 딕셔너리를 만들었을 때 

그것을 따로 저장하지 않고 나중에 다시 작업을 하면

컴퓨터가 코딩작업을 다시 하게 되어서 기다리는 시간이 생기게 됩니다

시간을 아끼고 작업을 원활하게 하기 위해서

그리고 결과를 보존하려면 모든 객체를 저장할 수 있는 pickle을 이용하면 됩니다.

 

저장하기

예로 pickle를 이용해 딕셔너리를 저장해보겠습니다. 

>>> import pickle
>>> data_dict = {}
>>> data_dict['A'] = [0,1,2]
>>> data_dict['B'] = [3,4,5]
>>> data_dict['C'] = [6,7,8]

>>> data_dict
{'A': [0, 1, 2], 'B': [3, 4, 5], 'C': [6, 7, 8]}

# Save data_dict
>>> with open('data_dict.pkl','wb') as f:
...    pickle.dump(data_dict,f)

pickle.dump로 저장을 하게 되면 해당 폴더에 다음과 같은 파일이 생깁니다.

 

 

이렇게 pkl 파일이 생겼다면 저장이 잘 된 것입니다.

 

불러오기

다시 읽어오도록 하겠습니다.

>>> with open('data_dict.pkl','rb') as f:
...    mydict = pickle.load(f)
   
>>> mydict
{'A': [0, 1, 2], 'B': [3, 4, 5], 'C': [6, 7, 8]}

mydict으로 불러왔습니다. mydict을 보면 제가 저장했던 딕셔너리가 불러온 것을 볼 수 있습니다.

 

이렇게 pickle을 이용하면 간단하게 저장하고 불러올 수 있습니다. 

다만, pkl 파일은 바이너리 파일 형식이므로 반드시 'rb', 'wb'로 읽고 써야하는 걸 명심하세요~!