2010-04-01 62 views
14

我有一个Python对象列表。然后我有另一个对象列表。我想通过第一个列表,看看是否有任何项目出现在第二个列表中。Python检查对象是否在对象列表中

我想我可以简单地做

for item1 in list1: 
    for item2 in list2: 
     if item1 == item2: 
      print "item %s in both lists" 

然而,这似乎并没有工作。虽然如果我这样做:

if item1.title == item2.title: 

它的作品没关系。我有比这更多的属性,尽管如此,如果我不需要比较所有的属性,那么我不想做大的if语句。

任何人都可以给我帮助或建议我可以做什么来找到出现在两个列表中的对象。

谢谢

回答

24

假设你的对象只有一个title属性,是相关的平等,就必须实现__eq__方法如下:

class YourObject: 
    [...] 
    def __eq__(self, other): 
     return self.title == other.title 

当然,如果你有更多的属性,这些属性是相关的平等,你也必须包括这些。您也可以考虑实施__ne____cmp__以获得一致的行为。谁出现在两个列表

5

set intersection将做到这一点。

>>> x=[1,2,3,4] 
>>> y=[3,4,5,6] 
>>> for i in set(x) & set(y): 
...  print "item %d in both lists" %i 
... 
item 3 in both lists 
item 4 in both lists 
+3

我认为他的问题是对象平等,而不是发现:) – extraneon 2010-04-01 08:39:47

+3

OP有对象列表,而不是原子类型列表。如果你使用没有定义'__hash__'的对象来尝试你的代码,它将不起作用,就像OP的代码不能处理没有定义'__eq__'或'__cmp__'的对象列表一样。 – hughdbrown 2010-04-01 15:05:44

9

如果对象是不一样的情况下,你需要实现__eq__方法蟒蛇是能够告诉当2个对象实际上是相等的。

当然,大多数库类型(例如字符串和列表)已经实现了__eq__,这可能是比较标题对您有用的原因(它们是字符串吗?)。

欲了解更多信息,请参阅python documentation
这是__eq__random example

+0

是的,他们不是同一个例子。你可以给我一个__eq__函数的例子,因为我对Python很陌生 – John 2010-04-01 08:41:23

+0

@John - 我已经添加了一个示例链接。只要搜索谷歌代码搜索“def __eq__”为我解决了它:) – abyx 2010-04-01 08:46:02

4

寻找对象:

l1 = [1,2,3,4,5] 
l2 = [3,4,5] 
common = set(l1).intersection(set(l2)) 

__eq__实施对象上结合这为其他建议。

0

尝试以下操作:

list1 = [item1, item2, item3] 
list2 = [item3, item4, item5] 
for item in list1: 
    if item in list2: 
     print "item %s in both lists" % item 
2
matches = [x for x in listA if x in listB] 
+0

除非我误解'in'是如何工作的,其运行时间将是列表长度的产物,这可能是不好的。 – shabbychef 2010-04-01 18:50:06

3

你需要写一个__eq__功能来定义如何比较对象是否相等。如果你想排序,那么你应该有一个__cmp__函数,并且按照__cmp__来实现__eq__是最有意义的。

def __eq__(self, other): 
    return cmp(self, other) == 0 

你或许应该还实现__hash__,你绝对应该,如果你打算把你的对象为一组或字典。对象的默认__hash__是id(),它有效地使所有对象都是唯一的(即唯一性不基于对象内容)。

我为一种类型的等价比较写了一个基类/接口。你可能会发现它有用:

class Comparable(object): 
    def attrs(self): 
     raise Exception("Must be implemented in concrete sub-class!") 
    def __values(self): 
     return (getattr(self, attr) for attr in self.attrs()) 
    def __hash__(self): 
     return reduce(lambda x, y: 37 * x + hash(y), self.__values(), 0) 
    def __cmp__(self, other): 
     for s, o in zip(self.__values(), other.__values()): 
      c = cmp(s, o) 
      if c: 
       return c 
     return 0 
    def __eq__(self, other): 
     return cmp(self, other) == 0 
    def __lt__(self, other): 
     return cmp(self, other) < 0 
    def __gt__(self, other): 
     return cmp(self, other) > 0 

if __name__ == '__main__': 
    class Foo(Comparable): 
     def __init__(self, x, y): 
      self.x = x 
      self.y = y 
     def attrs(self): 
      return ('x', 'y') 
     def __str__(self): 
      return "Foo[%d,%d]" % (self.x, self.y) 

    def foo_iter(x): 
     for i in range(x): 
      for j in range(x): 
       yield Foo(i, j) 

    for a in foo_iter(4): 
     for b in foo_iter(4): 
      if a<b: print "%(a)s < %(b)s" % locals() 
      if a==b: print "%(a)s == %(b)s" % locals() 
      if a>b: print "%(a)s > %(b)s" % locals() 

派生类必须实现attrs()返回有助于其身份(即不变的属性,使得它它是什么)对象的属性的元组或列表。最重要的是,代码在存在多个属性的情况下正确处理等同性,这是旧学校代码,通常不正确。

相关问题