如何使我的类实例可用作字典键？

Question

class A():
   def __init__(self, data=''):
       self.data = data  

   def __str__(self):
       return str(self.data)

d = {}  
elem = A()  
d[elem] = 'abc'  

elem2 = A()
print d[elem2]    # KeyError  
# actually elem2! was used not elem

我该怎么做才能不出错呢？

我试着用另一个A()的实例来获取 d[elem2]（不是 elem），但是内容是一样的。

Answer 1

你所做的应该是可以的，只要你不重写 __hash__() 和 __eq__() 这两个方法。这样的话，它会根据对象的身份来判断是否相等。如果你想要不同的相等标准，可以重写你类中的 __hash__() 和 __eq__() 方法。

Answer 2

答案是肯定的，你需要重新定义一下 __hash__() 和 __eq__()：

>>> class A(object):
...   def __init__(self, data=''):
...     self.data = data
...   def __eq__(self, another):
...     return hasattr(another, 'data') and self.data == another.data
...   def __hash__(self):
...     return hash(self.data)
... 
>>> a1, a2, a3 = A('foo'), A('foo'), A('bar')
>>> d = {a1: 'foo'}
>>> d[a1]
'foo'
>>> d[a2]
'foo'
>>> d[a3]
Traceback (most recent call last):
  File "", line 1, in 
KeyError: __main__.A object at 0x927d0>

正如另一个评论中所说，默认的 __hash__ 实现其实就是简单的身份识别，所以如果你想让它更复杂一些，就需要明确地定义它。