python set.__contains__ 的意外行为

Question

借用 __contains__ 文档中的文档

print set.__contains__.__doc__
x.__contains__(y) <==> y in x.

这对于原始对象（如 int、basestring 等）似乎工作得很好。但对于定义 __ne__ 和 的用户定义对象>__eq__ 方法，我得到了意想不到的行为。这是示例代码：

class CA(object):
  def __init__(self,name):
    self.name = name

  def __eq__(self,other):
    if self.name == other.name:
      return True
    return False

  def __ne__(self,other):
    return not self.__eq__(other)

obj1 = CA('hello')
obj2 = CA('hello')

theList = [obj1,]
theSet = set(theList)

# Test 1: list
print (obj2 in theList)  # return True

# Test 2: set weird
print (obj2 in theSet)  # return False  unexpected

# Test 3: iterating over the set
found = False
for x in theSet:
  if x == obj2:
    found = True

print found   # return True

# Test 4: Typcasting the set to a list
print (obj2 in list(theSet))  # return True

那么这是一个错误还是一个功能？

Answer 1

对于 set 和 dicts，您需要定义 __hash__。任何两个相等的对象都应该具有相同的哈希值，以便在 set 和 dicts 中获得一致/预期的行为。

我建议使用 _key 方法，然后在需要比较项目部分的任何地方引用该方法，就像您从 __ne____eq__ 一样code> 而不是重新实现它：

class CA(object):
  def __init__(self,name):
    self.name = name

  def _key(self):
    return type(self), self.name

  def __hash__(self):
    return hash(self._key())

  def __eq__(self,other):
    if self._key() == other._key():
      return True
    return False

  def __ne__(self,other):
    return not self.__eq__(other)

Answer 2

这是因为 CA 没有实现 __hash__ ：

一个合理的实现是

def __hash__(self):
    return hash(self.name)

Answer 3

集合散列它的元素以允许快速查找。您必须覆盖 __hash__ 方法，以便可以找到元素：

class CA(object):
  def __hash__(self):
    return hash(self.name)

列表不使用散列，而是像 for 循环一样比较每个元素。