[python/pickle] pickle data load/read, write

dictionary, list, np arr같은 python 객체를 파일로 binary파일로 저장하여 빠른 속도 처리가 가능하도록 함.

 

pkl file reading

# Data Loading
import pickle
import os

data_dir = './dataset/pkl_dir/'
file_lst = os.listdir(data_dir) # directory안에 있는 모든 pkl파일을 다 로딩해서 그 중 하나만 load

with open(os.path.join(data_dir, file_lst[0]), 'rb') as fr: # 여기선 0-index(첫번째)파일을 로드
    data = pickle.load(fr)

print(data)
print(data.keys())

 

pkl file writing

# Data Writing
import pickle
import os
import numpy as np

data = ['1', '2', '3'] # list 형식
data = np.zeros(255) # np array 형식
data = {'1': [0, 0], '2': [4, 5]} # dict 형식

with open("./test.pkl", "wb") as fw:
    pickle.dump(data, fw)