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

Question

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

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

但保存时必须手动强制执行，即：

^{pr2}$

有没有用变量设置NPZ字典键？i、 e.类似的东西

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

Answer 1

使用字典，您可以：

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

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