python - 我可以在用 savez 保存时注释一个 numpy 数组吗

Question

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

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

Answer 1

简而言之，您可以(.npz 只是一个 pickled dict)但您可能最好还是换成别的东西。 (如果你确实想坚持使用 .npz，看起来 @JoshAdel 刚刚发布了一个很好的例子。)

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()