如何比较python中的两个列表并返回匹配

我想取两个列表，并找出出现在这两个列表中的值。

a = [1, 2, 3, 4, 5]
b = [9, 8, 7, 6, 5]

returnMatches(a, b)

例如，将返回[5]。

当前回答

使用__and__属性方法也可以。

>>> a = [1, 2, 3, 4, 5]
>>> b = [9, 8, 7, 6, 5]
>>> set(a).__and__(set(b))
set([5])

或者简单地

>>> set([1, 2, 3, 4, 5]).__and__(set([9, 8, 7, 6, 5]))
set([5])
>>>

其他回答

使用set.intersection()，它快速且可读。

>>> set(a).intersection(b)
set([5])

可以使用itertools。产品。

>>> common_elements=[]
>>> for i in list(itertools.product(a,b)):
...     if i[0] == i[1]:
...         common_elements.append(i[0])

另一种更实用的方法是检查列表1 (lst1)和列表2 (lst2)是否相等，其中对象的深度为1，并保持顺序:

all(i == j for i, j in zip(lst1, lst2))

最简单的方法是使用set:

>>> a = [1, 2, 3, 4, 5]
>>> b = [9, 8, 7, 6, 5]
>>> set(a) & set(b)
set([5])

您也可以尝试这样做，将公共元素保存在一个新列表中。

new_list = []
for element in a:
    if element in b:
        new_list.append(element)

推荐文章