使用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()