python - 将字典保存为 .npy 格式

Question

我有一个 dict，并且我将一个数组与每个键相关联(键本身是一个数字)。

小例子:

import numpy as np
data = {}
data[2.5] = np.array([np.array([1,2,3,4]), np.array([5,6,7,8])])

然后我保存 dict:

np.save('file.npy', data)

然后重新加载它:

datanew = np.load('file.npy')

--

现在，为了访问每个 key 中存储的内容，我不能这样做:

datanew[2.5]

但我必须这样做

datanew[()][2.5]

• 为什么？
• 有没有更好的方法来保存字典？

Answer 1

原因是因为np.save的 arr 参数需要一个数组。当您传递字典时，它会将其保存为一个一个“无量纲”数组。因此，当您加载它时，您需要从该无量纲数组中获取“第一个”元素(即 [()])。你可以在调用 np.load 时这样做，然后再也不用担心它了:

datanew = np.load('file.npy')[()]

或者，由于您要保存字典，因此可以使用 pickle。 np.save 应该针对数值数组进行了优化，但我不知道您是否仍然可以获得将数组放入字典中的好处...