如何找到列表交集?

a = [1,2,3,4,5]
b = [1,3,5,6]
c = a and b
print c

实际输出:[1,3,5,6] 预期输出:[1,3,5]

如何在两个列表上实现布尔AND操作(列表交集)?

当前回答

对我来说，使用列表推导式是一个非常明显的方法。不确定性能如何，但至少能保持列表。

[x for x in a if x in b]

或者"所有在A中的x值，如果x值在B中"

其他回答

如果顺序不重要，你不需要担心重复，那么你可以使用set intersection:

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

使用过滤器和lambda运算符可以实现函数式的方法。

list1 = [1,2,3,4,5,6]

list2 = [2,4,6,9,10]

>>> list(filter(lambda x:x in list1, list2))

[2, 4, 6]

编辑:它过滤掉了同时存在于list1和list中的x，集差异也可以使用:

>>> list(filter(lambda x:x not in list1, list2))
[9,10]

python3 filter返回一个过滤器对象，用list封装它返回输出列表。

当我们使用tuple时，我们想要交叉

a=([1,2,3,4,5,20], [8,3,9,5,1,4,20])

for i in range(len(a)):

    b=set(a[i-1]).intersection(a[i])
print(b)
{1, 3, 4, 5, 20}

对我来说，使用列表推导式是一个非常明显的方法。不确定性能如何，但至少能保持列表。

[x for x in a if x in b]

或者"所有在A中的x值，如果x值在B中"

你也可以使用numpy.intersect1d(ar1, ar2)。它返回两个数组中唯一且已排序的值。

推荐文章