python:我的类作为dict键。怎么样?

时间:2011-03-07 15:07:24

标签: python dictionary key

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()BUT实例获取d[elem2](不是elem)。 (对我感到羞耻)
还是..我怎么能这样做?重新定义__hash__

2 个答案:

答案 0 :(得分:10)

答案是肯定的,您需要重新定义__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__的默认实现只是简单的身份,所以如果你想让它更复杂,你需要明确地定义它。

答案 1 :(得分:4)

只要您不覆盖__hash__()__eq__()方法,您所做的就应该有效。它将使用对象标识作为相等。如果您想要一个不同的平等概念,可以覆盖您班级的__hash__()__eq__()方法。