使用 numpy savez 时使用变量作为键名

Question

加载 npz 文件后，我喜欢能够使用键访问数组，例如：

KEY1  = "names"
file  = np.load(npzFilename)
data  = file[KEY1]

但你必须在保存时手动强制执行此操作，即：

np.savez(npzFilename, names=names)

有没有办法使用变量设置 NPZ 字典键？即类似的东西

np.savez(npzFilename, names, key=KEY1)

Answer 1

使用字典你可以这样做：

vals_to_save = {KEY1:names}
np.savez(npzFilename, **vals_to_save)

您可以根据需要以编程方式设置字典vals_to_save。