pickle()-파일,리스트 저장

pickle

• 파이썬에서 작업중이던 리스트, 딕셔너리 등을 저장해서 다른 위치에서 열고 싶을 경우
• 피클 파일을 이용하여 저장, 불러오기가 가능합니다.
• with open(파일 이름, 파일 모드) as f: # <- f라는 이름의 파일로 파일 이름, 모드를 연다는 의미.

import pickle

list_save = [1, 2, 3, 4]

# 피클 파일로 저장하는 겁니다. 파일 명은 "list_ex.pkl"로 저장
# 편집 가능하게 'wb'
with open("list_ex.pkl","wb") as f:
    pickle.dump(list_save, f)

# 불러오는 방법입니다
import pickle

with open("list_save.pkl","rb") as f:
    list_load = pickle.load(f)

print(list_load)
# [1, 2, 3, 4]

• 파일 모드 설명(보통 wb,rb를 많이 사용))

|file_mode|기능|설명|
|’r’|읽기 전용|파일을 읽어오는 기능이며, 파일이 없으면 에러|
|’w’|쓰기 전용|파일이 있으면 내용을 덮어 씀|
|’a’|추가|파일이 없으면 파일을 생성|
|’b’|바이너리 모드|파일의 내용을 그대로 읽고, 값을 그대로 사용|

joblib으로 저장, 불러오기

#joblib으로 dump하여 피클 형식으로 저장하기.
import joblib

joblib.dump((x_train, x_test, y_train, y_test), 'review.pkl')

# 불러오기
x_train, x_test, y_train, y_test = joblib.load('review.pkl')