如何比較python中的兩個列表並返回匹配結果

Question

我想採用兩個列表並查找兩個列表中出現的值。

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

returnMatches(a, b) 

例如將返回[5]。

Answer 1

不是最有效的，但目前爲止最明顯的方式做到這一點是：

>>> [i for i, j in zip(a, b) if i == j] 
[5] 

：

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

如果順序顯著你可以用列表理解這樣做（只適用於大小相同的列表，這些順序意義暗示着）。

Answer 2

做到這一點的最簡單的方法是使用sets：

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

Answer 3

使用set.intersection()，它的快速性和可讀性。

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

Answer 4

快速路：

list(set(a).intersection(set(b))) 

Answer 5

你要重複？如果沒有，也許你應該使用組，而不是：

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

Answer 6

您可以使用

def returnMatches(a,b): 
     return list(set(a) & set(b)) 

Answer 7

>>> s = ['a','b','c'] 
>>> f = ['a','b','d','c'] 
>>> ss= set(s) 
>>> fs =set(f) 
>>> print ss.intersection(fs) 
    **set(['a', 'c', 'b'])** 
>>> print ss.union(fs)   
    **set(['a', 'c', 'b', 'd'])** 
>>> print ss.union(fs) - ss.intersection(fs) 
    **set(['d'])** 

Answer 8

我更喜歡基於集合的答案，但這裏有一個工程反正

[x for x in a if x in b] 

Answer 9

而且你可以通過在一個新列表中保存常用元素來嘗試這一點。

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

Answer 10

顯示盧茨的解決方案快速的性能測試是最好的：

import time 

def speed_test(func): 
    def wrapper(*args, **kwargs): 
     t1 = time.time() 
     for x in xrange(5000): 
      results = func(*args, **kwargs) 
     t2 = time.time() 
     print '%s took %0.3f ms' % (func.func_name, (t2-t1)*1000.0) 
     return results 
    return wrapper 

@speed_test 
def compare_bitwise(x, y): 
    set_x = frozenset(x) 
    set_y = frozenset(y) 
    return set_x & set_y 

@speed_test 
def compare_listcomp(x, y): 
    return [i for i, j in zip(x, y) if i == j] 

@speed_test 
def compare_intersect(x, y): 
    return frozenset(x).intersection(y) 

# Comparing short lists 
a = [1, 2, 3, 4, 5] 
b = [9, 8, 7, 6, 5] 
compare_bitwise(a, b) 
compare_listcomp(a, b) 
compare_intersect(a, b) 

# Comparing longer lists 
import random 
a = random.sample(xrange(100000), 10000) 
b = random.sample(xrange(100000), 10000) 
compare_bitwise(a, b) 
compare_listcomp(a, b) 
compare_intersect(a, b) 

這是我的機器上的結果：

# Short list: 
compare_bitwise took 10.145 ms 
compare_listcomp took 11.157 ms 
compare_intersect took 7.461 ms 

# Long list: 
compare_bitwise took 11203.709 ms 
compare_listcomp took 17361.736 ms 
compare_intersect took 6833.768 ms 

顯然，任何人爲的性能測試應採取一粒鹽，但由於set().intersection()的答案是至少與其他解決方案一樣快，也是最readab樂，它應該是這個常見問題的標準解決方案。

Answer 11

您可以使用：

a = [1, 3, 4, 5, 9, 6, 7, 8] 
b = [1, 7, 0, 9] 
same_values = set(a) & set(b) 
print same_values 

輸出：

set([1, 7, 9]) 

Answer 12

另一個更實用一點的方法來檢查清單1（LST1）和表2（LST2）名單平等，其中的對象具有深度一個和它保持的順序是：

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

Answer 13

也可以使用itertools.product。

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

Answer 14

如果你想有一個布爾值：

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

Answer 15

使用__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]) 
>>>  

Answer 16

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

lista =set(a) 
listb =set(b) 
print listb.intersection(lista) 
returnMatches = set(['5']) #output 

print " ".join(str(return) for return in returnMatches) # remove the set() 

5  #final output