pickle

파이썬으로 프로그램을 작성하고 실행을 할 때 많은 변수들이 생겼다가 사라지기도 하고 변수의 값도 많은 변동이 있다.

또한 프로그램은 실행이 끝나면 모든 데이터가 사라진다.

이때 이 데이터를 저장을 했다가 다시 사용을 한다거나 다른 파일에서 이 데이터를 그대로 사용하고 싶을 때.

 

바로 pickle이 이것을 충족시켜준다.

 

pickle은 프로그램에서 사용하고 있는 데이터를 파일 형태로 저장하거나 불러올 수 있게 해주는 파이썬의 모듈이다.

 

사용법

import pickle

먼저 pickle 모듈을 import해준다.

 

 

1. 데이터 파일 생성하기

import pickle

데이터파일변수 = open("파일명.pickle", "wb")

파일형식은 .pickle로 만들어주고 쓰기모드인 w를 사용하는데 이때 pickle로 저장할 파일은 텍스트가 아닌

바이너리형식이다.

(이미지 파일, 음악파일, 실행 파일 등 -> 바이너리 파일)

그렇기때문에 w뒤에 b를 붙여서 'wb'로 써주어야 올바르게 저장이 된다.

(텍스트파일과 다르게 데이터에 한글이 있어도 별도의 인코딩 지정이 불필요!)

 

pickle에서 데이터를 파일로 저장할 때 dump()메소드를 사용한다.

dump(저장할 데이터, 데이터파일변수)

 

import pickle

FIFA_ability_file = open("ability.pickle", "wb") #바이너리타입지정
ability = {"이름":"손흥민","오버롤":89, "속도":88, "드리블":86, "슈팅":87, "패스":82}

pickle.dump(ability, FIFA_ability_file)
FIFA_ability_file.close() #파일 닫기

ㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡ

※주의할점!!

pickle.py로 파이썬파일을 저장해서 사용하면 오류가 발생한다.

이는 사용하는 모듈과 동일한 이름으로 파일을 생성하면

import 시에 동일한 경로에 있는 파일을 먼저 인식하게 되기 때문이라고 한다! (by_나도코딩 선생님)

ㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡ

 

 

2. 데이터 파일 불러오기

import pickle

데이터파일변수 = open("파일명.pickle", "rb")

읽기모드에서도 역시 'rb'로 바이너리타입을 지정해주어야 한다.

 

데이터를 불러올 때에는 load메소드를 사용한다.

load(데이터파일변수)

import pickle

FIFA_ability_file = open("ability.pickle", "rb") #바이너리타입지정
ability = pickle.load(FIFA_ability_file)
print(ability)
FIFA_ability_file.close()
ㅡㅡㅡㅡㅡㅡㅡㅡ
{"이름":"손흥민","오버롤":89, "속도":88, "드리블":86, "슈팅":87, "패스":82}

위와 같이 데이터를 불러와서 변수에 잘 저장된 것을 알 수 있다.

 

 

 

+Bonus

혼자서 실험해본 pickle에 여러 데이터 넣기

 

import pickle

1. 파일에 저장할 때

profile_file = open("profile.pickle", "wb")
profile1 = {"이름":"김점례", "나이":20, "취미":["독서","골프"]}
profile2 = {"이름":"김말숙", "나이":23, "취미":["축구","게임"]}

pickle.dump(profile1, profile_file)
pickle.dump(profile2, profile_file) # profile에 있는 정보를 파일에 저장
profile_file.close()

#-------------------------------------------

2. 파일 내용 불러오기

profile_file = open("profile.pickle", "rb")
profile1 = pickle.load(profile_file)
profile2 = pickle.load(profile_file) # 파일 내용 불러오기
print(profile1)
print(profile2)
profile_file.close()
ㅡㅡㅡㅡㅡㅡㅡㅡ
{"이름":"김점례", "나이":20, "취미":["독서","골프"]}
{"이름":"김말숙", "나이":23, "취미":["축구","게임"]}

#-------------------------------------------

2-1. 파일 내용 반복문으로 불러오기

profile_file = open("profile.pickle", "rb")
while True:
    try:
        profile = pickle.load(profile_file)
        if not profile:
            break
        print(profile)
    except EOFError: #데이터파일이 비어있으면 오류가 발생해서 예외처리
        pass
profile_file.close()
ㅡㅡㅡㅡㅡㅡㅡㅡ
{"이름":"김점례", "나이":20, "취미":["독서","골프"]}
{"이름":"김말숙", "나이":23, "취미":["축구","게임"]}

pickle에 대해서 추가적으로 알게된 것들이 있으면 추가할 예정! (21.9.28)