使用pickle可以把變量按照原來的形式存入硬盤,不管是python自帶的list、dict…結構,還是numpy的矩陣結構,各種數據結構都可以。
import pickle
import numpy as np
your_data = np.ones((4,5),np.uint8)
# your_data = [np.ones((4,5), np.uint8), np.ones((8,9), np.int64)] 什麼結構都可以
# wb 以二進制寫入
data_output = open('data.pkl','wb')
pickle.dump(your_data,data_output)
data_output.close()
# rb 以二進制讀取
data_input = open('data.pkl','rb')
read_data = pickle.load(data_input)
data_input.close()