자료형 그대로 파일에 저장하기

자료형 그대로 파일에 저장하기

원하는 자료형을 파일에 저장하려면 pickle 모듈을 사용하면 된다.

먼저 pickle 모듈을 불러온다.

import pickle

파이썬의 자료구조형인 튜플, 리스트, 딕셔너리가 잘 저장이 되는지 확인하기 위해 튜플, 리스트, 딕셔너리 자료구조를 구현하고 확인해 보았다.

tuple = ('a', 'b', 'c')
list = ['a', 'b', 'c']
dict = {'a':1, 'b':2, 'c':3}

pickle 모듈로 저장하기 위해서 with 명령어를 사용하고 pickle.dump() 메서드를 사용해 튜플, 리스트, 딕셔너리를 파일에 저장한다.

pickle.dump() 메서드의 첫 번째 인자로는 저장하고 싶은 객체의 변수명을 넣고 두 번째 인자로는 저장할 파일 경로를 넣어준다.

with open('./pickle', 'wb') as test:
    pickle.dump(tuple, test)
    pickle.dump(list, test)
    pickle.dump(dict, test)

파이썬으로 실행시켜보면 다음과 같은 바이너리 파일이 나온다.

바이너리로 된 파일을 읽기 위해서는 with open 명령어를 사용해 읽어 들이면 된다.

try except 문을 넣은 이유는 pickle.load() 메서드로 파일을 읽는데 파일의 내용이 더 이상 없으면 EOFError 예외를 던지기 때문에 처리하기 위해 넣어줬다.

pickle.load() 메서드의 인자로 파일의 경로를 넣어주면 된다.

with open('./pickle', 'rb') as test:
    try:
        while True:
            line = pickle.load(test)
            print(type(line), list)
    except EOFError:
        print('EOFError!!')

파이썬으로 실행시키면 tuple, list, dict 객체가 출력되는 것을 확인할 수 있다.