2个列表之间的常见元素比较

Question

def common_elements(list1, list2):
    """
    Return a list containing the elements which are in both list1 and list2

    >>> common_elements([1,2,3,4,5,6], [3,5,7,9])
    [3, 5]
    >>> common_elements(['this','this','n','that'],['this','not','that','that'])
    ['this', 'that']
    """
    for element in list1:
        if element in list2:
            return list(element)

到目前为止,但似乎无法让它工作!

有任何想法吗？

Answer 1

>>> list1 = [1,2,3,4,5,6]
>>> list2 = [3, 5, 7, 9]
>>> list(set(list1).intersection(list2))
[3, 5]

Answer 2

您还可以使用集合并在一行中获取共性:从包含其中一个集合的差异中减去包含该集合的集合.

A = [1,2,3,4]
B = [2,4,7,8]
commonalities = set(A) - (set(A) - set(B))

Answer 3

您可以使用numpy以下方法解决此问题：

import numpy as np

list1 = [1, 2, 3, 4, 5, 6]
list2 = [3, 5, 7, 9]

common_elements = np.intersect1d(list1, list2)
print(common_elements)

common_elements将是 numpy 数组：[3 5]。

Answer 4

所建议的解决方案S.Mark和SilentGhost通常告诉你应该如何在一个Python化的方式来完成,但我认为从知道为什么你的解决方案不工作,你也可能受益.问题是,只要找到两个列表中的第一个公共元素,就只返回该单个元素.您可以通过创建result列表并收集该列表中的公共元素来修复您的解决方案:

def common_elements(list1, list2):
    result = []
    for element in list1:
        if element in list2:
            result.append(element)
    return result

使用列表推导的更短版本:

def common_elements(list1, list2):
    return [element for element in list1 if element in list2]

但是,正如我所说,这是一种非常低效的方法 - Python的内置集类型更有效,因为它们在C内部实现.

Answer 5

使用set intersectionctions,set(list1)&set(list2)

>>> def common_elements(list1, list2):
...     return list(set(list1) & set(list2))
...
>>>
>>> common_elements([1,2,3,4,5,6], [3,5,7,9])
[3, 5]
>>>
>>> common_elements(['this','this','n','that'],['this','not','that','that'])
['this', 'that']
>>>
>>>

请注意,结果列表可能与原始列表的顺序不同.

Answer 6

你可以使用一个简单的列表理解:

x=[1,2,3,4]
y=[3,4,5]
common = [i for i in x if i in y]
common: [3,4]

Answer 7

我比较了每个答案提到的每种方法。目前我使用 python 3.6.3 来实现。这是我使用的代码：

import time
import random
from decimal import Decimal


def method1():
    common_elements = [x for x in li1_temp if x in li2_temp]
     print(len(common_elements))


def method2():
    common_elements = (x for x in li1_temp if x in li2_temp)
    print(len(list(common_elements)))


def method3():
    common_elements = set(li1_temp) & set(li2_temp)
    print(len(common_elements))


def method4():
    common_elements = set(li1_temp).intersection(li2_temp)
    print(len(common_elements))


if __name__ == "__main__":
    li1 = []
    li2 = []
    for i in range(100000):
        li1.append(random.randint(0, 10000))
        li2.append(random.randint(0, 10000))

    li1_temp = list(set(li1))
    li2_temp = list(set(li2))

    methods = [method1, method2, method3, method4]
    for m in methods:
        start = time.perf_counter()
        m()
        end = time.perf_counter()
        print(Decimal((end - start)))

如果您运行此代码，您可以看到，如果您使用列表或生成器（如果您迭代生成器，而不仅仅是使用它。当我强制生成器打印它的长度时，我这样做了），您将获得几乎相同的性能。但是如果你使用 set 你会得到更好的性能。此外，如果您使用相交方法，您将获得更好的性能。下面列出了我计算机中每种方法的结果：

1. 方法1：0.8150673999999999974619413478649221360683441
2. 方法2：0.8329545000000001531148541289439890533685684
3. 方法3：0.0016547000000000089414697868051007390022277
4. 方法4：0.0010262999999999244948867271887138485908508

Answer 8

之前的答案都可以找到唯一的共同元素,但无法解释列表中重复的项目.如果您希望公共元素显示在列表中常见的相同数字上,则可以使用以下单行:

l2, common = l2[:], [ e for e in l1 if e in l2 and (l2.pop(l2.index(e)) or True)]

or True只有在您期望评估任何元素时,才需要该部分False.

Answer 9

设置是我们可以解决此问题的另一种方法

a = [3,2,4]
b = [2,3,5]
set(a)&set(b)
{2, 3}