除了 pickle,对于 numpy 的 ndarray 数据,我们可以用 numpy 自带的 save/savez 以及 load 方法来序列化保存到硬盘上以及读取。
npy 文件可以保存一个 numpy 数据, 用的函数是 save
import numpy as np x = np.array([[1,2,3], [4,5,6]]) np.save('./123', x) y = np.load('./123.npy') print(y)
import numpy as np a = np.array([[1, 2, 3], [4, 5, 6]]) b = np.array([1, 2]) np.savez('/tmp/123.npz', a = a, b = b) data = np.load('/tmp/123.npz') print(data['a']) print(data['b']) data.close()
with np.load('foo.npz') as data: a = data['a']