python - 如何在循环中使用 `numpy.savez` 来保存多个数组？

Question

从一个循环中我得到了一个数组。我想将此数组保存在 tempfile 中。问题是 np.savez 只保存循环中的最后一个数组。我想我明白为什么会这样，但不知道如何做得更好。

为了解决我的问题，我想在 mode=a+b 中打开临时文件，目的是从循环中附加新数组。但这不起作用。

到目前为止我的代码:

tmp = TemporaryFile(mode="a+b")    
for i in range(10):
  array = getarray[i]  #demo purpose
  np.savez(tmp,array)
tmp.seek(0)

然后使用临时文件读取数组:

tmp_read = np.load(tmp)
print tmp_read.files
[OUTPUT]: ['arr_0']

但我想在临时文件中包含 10 个数组。有什么想法吗？
谢谢

Answer 1

您可以使用 *args 参数将多个数组保存在一个临时文件中。

np.savez(tmp, *getarray[:10])

或:

np.savez(tmp, *[getarray[0], getarray[1], getarray[8]])