Python:使用可哈希对象访问字典失败

Question

我正在使用可哈希对象作为字典的键。这些对象是可散列的，我可以将键值对存储在 dict 中，但是当我创建同一对象的副本时(给我相同的散列)，我得到一个 KeyError 。

下面是一些小示例代码:

class Object:
    def __init__(self, x): self.x = x
    def __hash__(self): return hash(self.x)

o1 = Object(1.)
o2 = Object(1.)
hash(o1) == hash(o2) # This is True
data = {}
data[o1] = 2.
data[o2] # Desired: This should output 2.

在我上面的场景中，如何实现 data[o2] 也返回 2.？

Answer 1

您需要同时实现__hash__ 和__eq__:

class Object:
    def __init__(self, x): self.x = x
    def __hash__(self): return hash(self.x)
    def __eq__(self, other): return self.x == other.x if isinstance(other, self.__class__) else NotImplemented

根据 Python documentation :

if a class does not define an __eq__() method it should not define a __hash__() operation either

找到哈希后，Python 的字典使用 __eq__ 比较键并意识到它们不同，这就是您没有得到正确输出的原因。