Python - 多个列表的交集？

Question

我正在玩 python，并且能够得到两个列表的交集：

result = set(a).intersection(b)

现在如果d 是一个包含a 和b 以及第三个元素c 的列表，那么是否有一个内置函数可以在d 中查找所有三个列表的交集？比如，

d = [[1,2,3,4], [2,3,4], [3,4,5,6,7]]

那么结果应该是

[3,4]

Answer 1

我发现reduce() 特别有用。其实numpy文档推荐使用reduce()来交叉多个列表：numpy.intersect1d reference

回答你的问题：

import numpy as np
from functools import reduce
# apply intersect1d to (a list of) multiple lists:
reduce(np.intersect1d, [list_1, list_2, ... list_n])

Answer 2

对于 2.4，你可以只定义一个交集函数。

def intersect(*d):
    sets = iter(map(set, d))
    result = sets.next()
    for s in sets:
        result = result.intersection(s)
    return result

---

对于较新版本的python：

intersection 方法接受任意数量的参数

result = set(d[0]).intersection(*d[1:])

或者，您可以将第一个集合与自身相交以避免分割列表并制作副本：

result = set(d[0]).intersection(*d)

我不太确定哪个会更有效，并且感觉它取决于 d[0] 的大小和列表的大小，除非 python 有类似的内置检查

if s1 is s2:
    return s1

在交集法中。

>>> d = [[1,2,3,4], [2,3,4], [3,4,5,6,7]]
>>> set(d[0]).intersection(*d)
set([3, 4])
>>> set(d[0]).intersection(*d[1:])
set([3, 4])
>>> 

Answer 3

您可以使用set.intersection(set1, set2, set3...) 获得任意数集的交集。因此，您只需要将列表转换为集合，然后将它们传递给此方法，如下所示：

d = [[1,2,3,4], [2,3,4], [3,4,5,6,7]]  
set.intersection(*[set(x) for x in d])  

结果：

{3, 4}

Answer 4

Lambda 减少。

from functools import reduce #you won't need this in Python 2
l=[[1, 2, 3, 4], [2, 3, 4], [3, 4, 5, 6, 7]]
reduce(set.intersection, [set(l_) for l_ in l])

Answer 5

@user3917838

很好很简单，但需要一些强制转换才能使其工作并作为结果给出一个列表。它应该看起来像：

list(reduce(set.intersection, [set(item) for item in d ]))

地点：

d = [[1,2,3,4], [2,3,4], [3,4,5,6,7]]

结果是：

[3, 4]

至少在 Python 3.4 中

Answer 6

set.intersection(*map(set,d))