将一些数组保存在同一文件中

Question

我想在同一文件中保存不同的数组：

a = [[1,2],[3,4],[5,6]]
b = [1,3,5]

我阅读了documentation关于np.savetxt函数的信息

尽管如此，我无法将a和b保存在同一文件中，以便在另一个程序中访问它们。

我想要这样的东西：

a = load("file_path",a)
b = load("file_path",b)

我该怎么做？

Answer 1

您可以改用np.savez

np.savez("file_path.npz", a=a, b=b)

然后加载

npzfile = np.load(outfile)

a = npzfile['a']
b = npzfile['b']

编辑：更新了np.savez调用，因此数组以自己的名称保存。