如何在Python中计算组合[重复]

Question

我想将组合集计算为从 n 项目中选择的 k 项目的所有可能子集的集合。比如n = 4和k = 3，那么项目是整数(0, 1, 2, 3)，那么就有4可能的组合元素：

[0, 1, 2]
[0, 1, 3]
[0, 2, 3]
[1, 2, 3]

在 Python 中是否有一个函数可以做到这一点？

Answer 1

itertools.combinations 产生所需的输出：

from itertools import combinations

n = 4
k = 3

# combinations() outputs tuples.
# Use list() to transform the tuples into lists.
print([list(combination) for combination in combinations(range(n), 3)])