比較2個列表中元素的有效方法

Question

我想比較2個列表中的條目：

例如：

for key in keys:
        for elem in elements:
            #Find a key in common with 2 lists and write the corresponding value
            if key == elem:
                 do something
            #Iterate through elements unit the current "key" value is found.
            else:
                 pass

這種方法的問題在於它非常慢，有沒有更快的方法呢？

*假設len(keys) > len(elements)

Answer 1

使用集合，然后集合相交將返回公共元素。

設置交集是O(min(len(s1), len(s2))操作。

>>> s1 = range(10)
>>> s2 = range(5,20)
>>> set(s1) & set(s2)  #set intersection returns the common keys
set([8, 9, 5, 6, 7])   # order not preserved 

Answer 2

將它們變成集合：

set1 = set(keys)
set2 = set(elements)

現在您可以找到他們的交集：

set1.intersection(set2)

這些集合不會保留順序，因此您將只獲得常見元素。