python：我的类是dict键。怎样？

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

我怎样才能不出错地实现这一点呢？

编辑：
FFFUUU，错误是：
我试图用A（）的另一个实例来获取d[elem2]（而不是elem），但内容相同。（我很惭愧）
还是。。我该怎么做？重新定义__hash__？

Answer 1

答案是肯定的，您需要重新定义__hash__()：

>>> 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__的默认实现只是简单的标识，所以如果您想使它更复杂，就需要显式地定义它。

Answer 2

只要不重写__hash__()和__eq__()方法，您所做的就应该有效。它将使用对象标识作为等式。如果您想要一个不同的相等概念，可以重写类的__hash__()和__eq__()方法。