找到多组交集的最佳方法？

Question

我有一个集合列表：

setlist = [s1,s2,s3...]

我想要 s1 ∩ s2 ∩ s3 ...

我可以编写一个函数来执行一系列成对的s1.intersection(s2)等。

有推荐的、更好的或内置的方法吗？

Answer 1

我相信最简单的做法是：

#assuming three sets
set1 = {1,2,3,4,5}
set2 = {2,3,8,9}
set3 = {2,10,11,12}

#intersection
set4 = set1 & set2 & set3

set4 将是 set1 、 set2 、 set3 的交集，并将包含值 2。

print(set4)

set([2])

Answer 2

Jean-François Fabre set.intesection(*list_of_sets) 答案绝对是最 Pyhtonic 并且正确地被接受的答案。

对于那些想使用reduce的人，以下也可以：

reduce(set.intersection, list_of_sets)

Answer 3

从 Python 2.6 版开始，您可以对set.intersection() 使用多个参数，例如

u = set.intersection(s1, s2, s3)

如果集合在列表中，则转换为：

u = set.intersection(*setlist)

*a_list 是 list expansion

请注意，set.intersection 不是静态方法，但它使用函数符号来应用第一个集合与列表其余部分的交集。因此，如果参数列表为空，这将失败。

Answer 4

显然set.intersection 是您想要的，但如果您需要“取所有这些的总和”、“取所有这些的乘积”、“取所有这些的异或”的概括，那么你要找的是reduce函数：

from operator import and_
from functools import reduce
print(reduce(and_, [{1,2,3},{2,3,4},{3,4,5}])) # = {3}

或

print(reduce((lambda x,y: x&y), [{1,2,3},{2,3,4},{3,4,5}])) # = {3}

Answer 5

在这里，我为多个集合交集提供了一个通用函数，试图利用可用的最佳方法：

def multiple_set_intersection(*sets):
    """Return multiple set intersection."""
    try:
        return set.intersection(*sets)
    except TypeError: # this is Python < 2.6 or no arguments
        pass

    try: a_set= sets[0]
    except IndexError: # no arguments
        return set() # return empty set

    return reduce(a_set.intersection, sets[1:])

Guido 可能不喜欢 reduce，但我有点喜欢它 :)

Answer 6

从 2.6 开始，set.intersection 可以任意多的迭代。

>>> s1 = set([1, 2, 3])
>>> s2 = set([2, 3, 4])
>>> s3 = set([2, 4, 6])
>>> s1 & s2 & s3
set([2])
>>> s1.intersection(s2, s3)
set([2])
>>> sets = [s1, s2, s3]
>>> set.intersection(*sets)
set([2])

Answer 7

如果您没有 Python 2.6 或更高版本，另一种方法是编写显式 for 循环：

def set_list_intersection(set_list):
  if not set_list:
    return set()
  result = set_list[0]
  for s in set_list[1:]:
    result &= s
  return result

set_list = [set([1, 2]), set([1, 3]), set([1, 4])]
print set_list_intersection(set_list)
# Output: set([1])

你也可以使用reduce:

set_list = [set([1, 2]), set([1, 3]), set([1, 4])]
print reduce(lambda s1, s2: s1 & s2, set_list)
# Output: set([1])

但是，很多 Python 程序员不喜欢它，including Guido himself：

大约 12 年前，Python 获得了 lambda、reduce()、filter() 和 map()，这归功于（我相信）一个错过了它们并提交了工作补丁的 Lisp 黑客。但是，尽管有 PR 值，我认为这些功能应该从 Python 3000 中删除。

所以现在减少（）。这实际上是我一直最讨厌的一个，因为除了一些涉及 + 或 * 的示例之外，几乎每次我看到带有非平凡函数参数的 reduce() 调用时，我都需要拿起笔和纸来在我理解 reduce() 应该做什么之前，先画出实际输入该函数的内容。所以在我看来，reduce() 的适用性几乎仅限于关联运算符，在所有其他情况下，最好明确地写出累加循环。