使用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)

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