Question

假设我的程序创建了大量数据，然后我用numpy的savez例程保存。但是，我还想将一些额外的信息与该数组一起存储。例子是当前版本的git commit id，以及用于生成数据的输入参数，以便稍后我可以查看数据并确切知道我是如何创建数据的。

有没有办法将这些信息与数组一起直接保存在npz文件中，还是我必须创建一个单独的文件？

Answer 1

简而言之，你可以（.npz只是一个腌制的字典），但你可能最好切换到别的东西。（看起来@JoshAdel刚刚发布了一个很好的例子，如果你想坚持使用.npz。）

对于像这样的事情，HDF是一个更好的选择。

hdf文件中的每个组或数据集都可以存储属性。

我推荐h5py将numpy数组存储在hdf文件中。

举个例子：

import numpy as np
import h5py

somearray = np.random.random(100)

f = h5py.File('test.hdf', 'w')

dataset = f.create_dataset('my_data', data=somearray)

# Store attributes about your dataset using dictionary-like access
dataset.attrs['git id'] = 'yay this is a string'

f.close()

Answer 2

你应该能够：

In [2]: a = np.arange(10)

In [3]: b = 'git push'

In [5]: np.savez('file',a=a,b=b)

In [7]: data = np.load('file.npz')

In [8]: data.keys()
Out[8]: ['a', 'b']

In [9]: data['a']
Out[9]: array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

In [10]: str(data['b'])
Out[10]: 'git push'

因此，您可以保存任意命名数据并获取类似字典的对象。使用的更好的格式可能更灵活，并且内置支持各种元数据的是使用h5py或pytables的hdf5：

http://h5py.alfven.org/docs/

http://www.pytables.org/

使用savez保存时，我可以注释一个numpy数组吗？

2 个答案: