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

时间:2016-10-07 17:38:25

标签: python list object set difference

我的目标是区分包含对象的两个列表。

我已经实现了一个名为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

但是我希望仅将{3}作为b1b1b的输出作为输出。

是否可以使用套装来解决这个问题?或者我应该从不同的角度处理问题吗?

1 个答案:

答案 0 :(得分:4)

您需要实现哈希,您选择的是由您决定的,但以下内容可行:

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))