python - Python中使用哈希获取两个对象列表之间的差异

Question

我的目标是获取两个包含对象的列表之间的差异。

我已经实现了一个名为 Branch 的类并覆盖了它的 __eq__和__ne__方法如下:

class Branch(object):
    def __str__(self):
        return self.name

    def __eq__(self, other):
        if isinstance(other, Branch):
            return (self.valueFrom == other.valueFrom) \
                and (self.valueTo == other.valueTo) \
                and (self.inService == other.inService)
        return NotImplemented

    def __ne__(self, other):
        result = self.__eq__(other)
        if result is NotImplemented:
            return result
        return not result

    def __init__(self, name, valueFrom, valueTo, inService=True):
        self.name = name
        self.valueFrom = valueFrom
        self.valueTo = valueTo
        self.inService = inService

我的第一次尝试是使用方法 difference来自set类型。然而，这似乎是不可能的，因为它使用对象的哈希值而不是 __eq__方法如我所愿。

以下代码显示了问题:

b1 = Branch("branch1", 1, 2)
b1b = Branch("equal to branch1", 1, 2)
b2 = Branch("branch2", 2, 3)
b3 = Branch("branch3", 3, 1)
b3_off = Branch("branch3 not in service", 3, 1, False)

l1 =[b1,b2,b3]
l2 =[b1b,b2,b3_off]

difference = set(l1).difference(l2)
for branch in difference:
    print branch

输出是:

>>> 
branch1
branch3

但是我希望仅将branch3作为输出b1和b1b应该平等对待。

是否可以使用集合来解决这个问题？或者我应该从不同的角度来处理这个问题？

Answer 1

您需要实现哈希，您选择什么取决于您，但以下内容可以工作:

def __hash__(self):
    return hash((self.valueFrom , self.valueTo , self.inService))

您需要实现的只是 hash 和 eq:

class Branch(object):
    def __init__(self, name, valueFrom, valueTo, inService=True):
        self.name = name
        self.valueFrom = valueFrom
        self.valueTo = valueTo
        self.inService = inService

    def __eq__(self, other):
        if isinstance(other, Branch):
            return (self.valueFrom,self.valueTo,self.inService )\
                   ==(other.valueFrom, other.valueTo, other.inService)
        return NotImplemented

    def __str__(self):
        return self.name

    def __hash__(self):
        return hash((self.valueFrom, self.valueTo,self.inService))