如何实现__eq__进行集合包含测试?

时间:2013-03-10 20:09:21

标签: python set equality

我遇到了一个问题,我在一个集合中添加一个实例然后进行测试以查看该集合中是否存在该对象。我已经覆盖了__eq__(),但在包含测试期间没有调用它。我是否必须覆盖__hash__()?如果是这样,我将如何实现__hash__(),因为我需要散列元组,列表和字典?

class DummyObj(object):

    def __init__(self, myTuple, myList, myDictionary=None):
        self.myTuple = myTuple
        self.myList = myList
        self.myDictionary = myDictionary

    def __eq__(self, other):
        return self.myTuple == other.myTuple and \
            self.myList == other.myList and \
            self.myDictionary == other.myDictionary

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

if __name__ == '__main__':

    list1 = [1, 2, 3]
    t1    = (4, 5, 6)
    d1    = { 7 : True, 8 : True, 9 : True }
    p1 = DummyObj(t1, list1, d1)

    mySet = set()

    mySet.add(p1)

    if p1 in mySet:
        print "p1 in set"
    else:
        print "p1 not in set"

2 个答案:

答案 0 :(得分:11)

来自documentation on sets

  

使用词典实现集合类。因此,   set元素的要求与字典的要求相同   键;即,该元素定义__eq __()和__hash __()。

__hash__ function documentation建议将组件的散列放在一起。正如其他人所提到的,散列可变对象通常不是一个好主意,但如果你真的需要,这可行:

class DummyObj(object):

    ...

    def __hash__(self):
        return (hash(self.myTuple) ^
                hash(tuple(self.myList)) ^
                hash(tuple(self.myDictionary.items())))

检查它是否有效:

p1 = DummyObj(t1, list1, d1)
p2 = DummyObj(t1, list1, d1)
mySet = set()
mySet.add(p1)

print "p1 in set", p1 in mySet
print "p2 in set", p2 in mySet

打印:

$ python settest.py 
p1 in set True
p2 in set True

答案 1 :(得分:-3)

嗯,在使用'in'运算符比较对象时,python可能无法调用__eq____ne__。我不肯定具体的“富比较”运算符是什么,查看文档,但覆盖__cmp__应该解决您的问题,因为python默认使用它来执行对象比较,如果更合适的“丰富的比较”运算符未实现。