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