在Python中生成大小为k（包含k个元素）的所有子集

Question

我有一组值，并想创建包含2个元素的所有子集的列表。

例如，源集([1,2,3])具有以下2个元素的子集：

set([1,2]), set([1,3]), set([2,3])

有没有办法在python中做到这一点？

Answer 1

好像你想要的itertools.combinations：

>>> list(itertools.combinations((1, 2, 3), 2))
[(1, 2), (1, 3), (2, 3)]

如果要设置，则必须显式转换它们。如果您不介意使用迭代器而不是列表，并且使用的是Python 3，则可以使用map：

>>> s = set((1, 2, 3))
>>> map(set, itertools.combinations(s, 2))
<map object at 0x10cdc26d8>

要一次查看所有结果，可以将的输出传递map给list。（在Python 2中，的输出map自动为列表。）

>>> list(map(set, itertools.combinations(s, 2)))
[{1, 2}, {1, 3}, {2, 3}]

但是，如果您知道需要列表，则列表理解会略胜一筹（h / t JacobBowyer）：

>>> [set(i) for i in itertools.combinations(s, 2)]
[{1, 2}, {1, 3}, {2, 3}]