我的目标是区分包含对象的两个列表。
我已经实现了一个名为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}作为b1
和b1b
的输出作为输出。
是否可以使用套装来解决这个问题?或者我应该从不同的角度处理问题吗?
答案 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))