numpy中savez存储多个数组

最新推荐文章于 2025-10-08 10:10:35 发布

原创最新推荐文章于 2025-10-08 10:10:35 发布 · 4.5k 阅读

4 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

当前文章被以下社区和专栏收录：

这篇博客介绍了如何使用numpy的np.savez函数结合字典来存储多个数组。通过创建一个包含多个数组的字典，然后使用星号操作符将字典解压传递给np.savez，可以方便地保存多个数组到一个文件中。之后，通过np.load重新加载文件，可以轻松访问保存的数组。这种方法对于处理大量相关数组非常实用。

按照文档给的例子，使用np.savez只能以np.savez(filepath, a=array_a, b=array_b, …)的形式传参，如下所示。

np.savez(outfile, x, y)
_ = outfile.seek(0) # Only needed here to simulate closing & reopening file
npzfile = np.load(outfile)
npzfile.files
#['arr_0', 'arr_1']
npzfile['arr_0']
#array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

但有的时候需要传很多个数组，可以使用以下方法：
先用字典将数组存储起来，最后使用np.savez(filepath, **dict)存储，代码如下

a=np.arange(0,10)
b=np.arange(10,30)
array_dict = {'a':a, 'b':b}
np.savez('outfile.npz', **array_dict)

npzfile = np.load('outfile.npz')
npzfile.files
#['a', 'b']
npzfile['a']
#array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])