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

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

好像你想要的

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}]