可能重复:
2个列表之间的公共元素比较
我有两个列表:
[Apples, Bananas, Pears]
[Kiwis, Bananas, Apples]
Run Code Online (Sandbox Code Playgroud)
我想只获得两个列表共享的元素.在python中必须有一些内置函数.
结果:
[Apples, Bananas]
Run Code Online (Sandbox Code Playgroud)
Ale*_*yev 12
>>> set(["Apples", "Bananas", "Pears"]).intersection(["Kiwis", "Bananas", "Apples"])
set(['Bananas', 'Apples'])
>>> list(_)
['Bananas', 'Apples']
Run Code Online (Sandbox Code Playgroud)
这相当于:
>>>set(["Apples", "Bananas", "Pears"]) & set(["Kiwis", "Bananas", "Apples"])
set(['Bananas', 'Apples'])
Run Code Online (Sandbox Code Playgroud)
Mat*_*ams 10
原答案:
[x for x in list1 if x in list2]
Run Code Online (Sandbox Code Playgroud)
在回答关于保留订单的评论问题时:
列表理解只是一个简写版本
intersection = []
for x in list1:
if x in list2: #Note that using 'in' involves looping over list2
intersection.append(x)
Run Code Online (Sandbox Code Playgroud)
通过这个扩展版本,可以更容易地看到发生了什么.输出列表list1与要list2删除的元素完全相同.所以它将保持秩序list1.例如,如果list1 = [1,2,3]和list2 = [3,2,5],则列表推导的输出将是[2,3].如果列表的位置像这样颠倒了
[x for x in list2 if x in list1]
Run Code Online (Sandbox Code Playgroud)
那么顺序list2将保留在输出中,给我们[3,2].
此外,在可能不希望的副作用中,这意味着该方法将包含重复的元素list1.例如:
>>> [x for x in [1,2,3,3,3] if x in [2,3]]
[2, 3, 3, 3]
Run Code Online (Sandbox Code Playgroud)
所以对于你的例子,这发生了:
>>> [fruit for fruit in ["Apples", "Bananas", "Pears"] if fruit in ["Kiwis", "Bananas", "Apples"]]
['Apples', 'Bananas']
Run Code Online (Sandbox Code Playgroud)
但如果列表被翻转,则输出反转:
>>> [fruit for fruit in ["Kiwis", "Bananas", "Apples"] if fruit in ["Apples", "Bananas", "Pears"]]
['Bananas', 'Apples']
Run Code Online (Sandbox Code Playgroud)
因此,一般来说,集合解决方案更好,因为它更有效,因为通常您不需要重复元素.但是,如果您想保留订单,这是可行的方法.(如果要保留顺序并且没有重复项,则可以使用此方法然后去除重复项,具体取决于您是希望保留较早的副本还是后续副本.)