如何在 Numpy 中将数组字典保存到文件中

我有一个数组字典

{1:array([...]), 2:array([...]), 3:array([...])}

我想将其保存到文件中并稍后加载。

我发现 numpy 有一个列表输入输出 http://docs.scipy.org/doc/numpy/reference/routines.io.html方法，但似乎它们只处理数组。

Thanks.

以下脚本将把 numpy 数组的字典保存到磁盘，然后将其加载回内存。

import numpy as np

arr1 = np.arange(0, 10, 1)
arr2 = np.arange(10, 20, 1)
arr3 = np.arange(20, 30, 1)

dct = {'1': arr1, '2': arr2, '3':arr3}

outfile = 'dict_of_arrays.npz'
np.savez(outfile, **dct)

npzfile = np.load(outfile)
print('npzfile.files: {}'.format(npzfile.files))
print('npzfile["1"]: {}'.format(npzfile["1"]))

运行此脚本会显示以下内容：

npzfile.files: ['1', '3', '2']
npzfile["1"]: [0 1 2 3 4 5 6 7 8 9]

请注意，您的字典键必须是字符串。也许这就是问题所在？

我正在运行 numpy 1.10.4