使用pickle可以把变量按照原来的形式存入硬盘,不像写入到txt文本中,是以字符串的形式,读取之后还要转化,很麻烦。
import pickle
import numpy as np
your_data = np.ones((4,5),np.uint8)
# wb 以二进制写入
data_output = open('data.pkl','wb')
pickle.dump(your_data,data_output)
data_output.close()
# rb 以二进制读取
data_input = opne('data.pkl','rb')
read_data = pickle.load(data_input)
data_input.close()