python - numpy float64s 作为字典的键

Question

如果我有一个字典，其键是 numpy float64 数字，我该如何通过键值访问它们？

>>> keys = np.arange(0, 0.5, 0.05, dtype=np.float64)
>>> keys
    array([...,  0.3 ,  ...])

    # The following creates a dicionary lookup table
    # data[x] = exp(x) for all x in keys

>>> data = {key: np.exp(key) for key in keys}
>>> data[0.3]
    KeyError: 0.3
>>> data[np.float64(0.3)]
    KeyError: 0.29999999999999999
>>> data.keys()
[..., 0.30000000000000004, ...]

numpy float 甚至可以用作字典的键吗？

Answer 1

来自评论，“我正在做的真正计算不是每次你想要一个值时都必须做的事情。有没有更优雅的方法来创建 map ？”

这是一个标准问题，通用解决方案称为查找表 或 LUT。您可以通过搜索找到许多实现，有或没有 numpy 和 scipy。 Here's使用 scipy 的插值方法的解决方案。

正如其他人所说，使用 float 作为字典键是有问题的。此外，使用 LUT，您还可以在两个预先计算的解决方案之间进行插值，这可以提高准确性等。