[Pandas] 판다스 pickle 이용하기

pickle

• 파이썬의 변수, 함수, 객체를 파일로 저장하고 불러올 수 있느 라이브럴.
• list, dictionary 등을 파일 그대로 저장하면 용량이 매우 커지는데, pickle을 사용하면 binary 형태로 저장하기 때문에 용량이 매우 작아진다.
• gzip을 사용하면 pickle로 저장된 데이터를 압축할 수 있다.
• 머신러닝 모델 등을 저장하고 불러올 때에도 활용 가능

pickle 저장하기

import pandas as pd

# 샘플 데이터프레임 생성
data = {
    'Column1': [1, 2, 3, 4, 5],
    'Column2': ['A', 'B', 'C', 'D', 'E']
}
df = pd.DataFrame(data)

# 데이터프레임을 pickle 파일로 저장
df.to_pickle('dataframe.pkl')  # 데이터프레임을 'dataframe.pkl' 파일로 저장

pickle 불러오기

import pandas as pd

# pickle 파일에서 데이터프레임 불러오기
loaded_df = pd.read_pickle('dataframe.pkl')  # 'dataframe.pkl' 파일에서 데이터프레임 불러오기
print(loaded_df)