본문 바로가기

PYTHON/Pandas

input output

pickle 모듈

  • 파일 입출력 방법의 하나
  • 텍스트 이외의 자료형을 파일로 저장하기 위해 사용
  • 데이터를 자요형의 변겅없이 파일로 저장하여 그대로 로드 가능
  • 파이썬 객체에 대해 모두 적용 가능
  • import pickle
  • open('file.txt', 'rt') : 기본값으로 텍스트 읽기 모드
  • wb, rb : 바이트형식으로 읽거나 씀 .bin 확장자 사용
  • pickle.dump : 객체 저장
import pickle

with open("data.pkl", "rb") as f:       # data.pkl 파일을 바이너리 읽기 모드로 열기
    result = pickle.load(f)     # 한줄씩 읽어 오기

[참고] python_input output

dataframe pickle 저장 & 불러오기

df = pd.DataFrame({"a": [1, 2, 3], "b": [1, 3, 4]})
df.to_pickle("df.pkl")
df = pd.read_pickle("df.pkl")

csv 파일

  • comma-seperated values
  • 각 라인의 컬럼들이 콤마로 분리된 텍스트 파일 포맷
  • import csv
  • csv.reader() : iterator 타입인 reader 객체를 리턴하므로 컬럼을 나열한 리스트 타입으로 반환
  • csv.writer() : writerow로 추가

csv 저장 & 불러오기
# save
df.to_csv("datas", index=False)

# load
df = pd.read_csv("datas")

excel 저장
# save
df.to_excel("datas", sheet_name="Sheet1", engine="xlsxwriter", encoding="utf-8-sig", index=False)

# load
df = pd.read_excel("datas", "Sheet1")

'PYTHON > Pandas' 카테고리의 다른 글

excel 변환  (0) 2020.05.14
dataframe  (0) 2019.12.29
series  (0) 2019.12.29
basic  (0) 2019.12.29