python - 使用 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。