Python 데이터 저장/로딩 Pickle

파이썬은 자료 구조를 쉽게 파일로 저장 및 로딩할 수 있다.

+ Pickle : Save & Load        데이터 저장/로딩

주의! 저장한 순서대로 로딩해야 한다.

- pickle 모듈 사용

저장 파일이 있으면 로딩하고, 없으면 저장하는 샘플.

import os, pickle

savefile='mnist.pkl'

if os.path.exists(savefile) :

    print('loading from cache...')

    f = open(savefile, 'rb')

    x_train = pickle.load(f)

    x_test =  pickle.load(f)

    f.close()

else:

    print('downloading.. MNIST..')

    (x_train,_), (x_test, _) = mnist.load_data()

    f = open(savefile, 'wb')

    pickle.dump(x_train, f)

    pickle.dump(x_test, f)

    f.close()