python - 比较两个列表中元素的有效方法

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

使用集合，然后集合交集将返回公共(public)元素。

集合交集是一个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