如果我有这样的课程:
class foo(object):
def __init__(self, x, y, z):
self.x = x
self.y = y
self.z = z
在这样的列表中:
list = [foo(1, 2, 3), foo(4, 5, 6), foo(7, 8, 9)]
我如何为'in'创建自定义测试,以便仅检查x和z值,以便:
new_foo = foo(1,8,3)
if new_foo in list:
print True
else:
print False
会打印True
答案 0 :(得分:6)
使用in
列出使用相等性的测试,因此您需要定义__eq__
方法:请参阅the documentation。您还需要定义一个__hash__
方法,以确保您的对象在具有可变状态时以一致的方式进行比较。例如:
class foo(object):
def __init__(self, x,y,z):
self.x = x
self.y = y
self.z = z
def __eq__(self, other):
return (self.x, self.z) == (other.x, other.z)
def __hash__(self):
return hash((self.x, self.z))
但是,您应该仔细考虑是否真的想要这样做。它定义了一个平等概念,适用于所有测试平等的情况。因此,如果您在帖子中执行了所要求的内容,那么foo(1,2,3) == foo(1,8,3)
一般都是正确的,而不仅仅是在使用in
时。