itertools.combinations详解 - Python组合与概率计算

一、概述

itertools.combinations 是 Python 标准库 itertools 模块中的组合迭代器之一，用于从可迭代对象中生成所有长度为 r 的组合。与排列不同，组合不考虑元素的顺序，即 (A, B) 和 (B, A) 被视为同一个组合。

组合在统计学、概率计算、彩票分析、团队组建等场景中应用广泛。此外，combinations_with_replacement 允许同一元素在组合中多次出现，适用于可重复抽样的场景。

二、语法

普通组合（不允许重复选取）

itertools.combinations(iterable, r)

允许重复选取的组合

itertools.combinations_with_replacement(iterable, r)

三、参数说明

四、返回值

返回一个迭代器对象，产生所有长度为 r 的组合元组。组合按照输入可迭代对象中的元素顺序生成，每个组合中的元素保持输入中的相对顺序。结果总数为 C(n, r) = n! / (r! × (n-r)!)。

五、代码示例

示例1：基本组合

import itertools

# 从5个元素中取3个组合
combs = list(itertools.combinations(range(1, 6), 3))
print("C(5,3)组合:")
for c in combs:
    print(c)
print(f"总数: {len(combs)}")

# 输出：
# C(5,3)组合:
# (1, 2, 3)
# (1, 2, 4)
# (1, 2, 5)
# (1, 3, 4)
# (1, 3, 5)
# (1, 4, 5)
# (2, 3, 4)
# (2, 3, 5)
# (2, 4, 5)
# (3, 4, 5)
# 总数: 10

从 5 个元素中取 3 个组合，结果为 C(5,3) = 10 种。注意 (1, 2, 3) 和 (3, 2, 1) 是同一个组合，只出现一次。

示例2：彩票组合计算

import itertools

# 双色球：从33个红球中选6个
red_balls = range(1, 34)
red_combos = itertools.combinations(red_balls, 6)
total = sum(1 for _ in red_combos)
print(f"红球组合数: {total}")

# 从16个蓝球中选1个
blue_total = 16
print(f"蓝球组合数: {blue_total}")
print(f"双色球总组合数: {total * blue_total}")

# 输出：
# 红球组合数: 1107568
# 蓝球组合数: 16
# 双色球总组合数: 17721088

这个示例展示了 combinations 在概率计算中的应用。双色球的头奖概率约为 1772 万分之一。注意这里使用生成器表达式 sum(1 for _ in red_combos) 计数，避免将百万级数据加载到内存。

示例3：combinations_with_replacement

import itertools

# 普通组合 vs 允许重复的组合
items = 'ABC'

print("普通组合 C(3,2):")
for c in itertools.combinations(items, 2):
    print(c)

print("\n允许重复的组合 CR(3,2):")
for c in itertools.combinations_with_replacement(items, 2):
    print(c)

# 输出：
# 普通组合 C(3,2):
# ('A', 'B')
# ('A', 'C')
# ('B', 'C')
#
# 允许重复的组合 CR(3,2):
# ('A', 'A')
# ('A', 'B')
# ('A', 'C')
# ('B', 'B')
# ('B', 'C')
# ('C', 'C')

combinations_with_replacement 允许同一元素在组合中多次出现。普通组合 C(3,2) = 3 种，而允许重复的组合 CR(3,2) = 6 种，增加了 ('A','A')、('B','B')、('C','C') 三种情况。

六、实际应用场景

• 概率与统计：计算彩票中奖概率、抽样组合数、置信区间等

• 团队组建：从候选人中选出所有可能的团队组合

• 特征选择：在机器学习中，穷举所有特征子集进行特征选择

七、注意事项

注意1：组合数量随 n 和 r 的增大而快速增长。C(30, 15) 约为 1.55 亿，请评估计算和内存成本。

注意2：combinations 基于元素位置生成组合，如果输入中有重复元素，结果中会出现值相同的组合。需要去重时请使用 set()。

注意3：如果 r 大于输入元素个数，返回空迭代器；r 为 0 时返回一个包含空元组的迭代器。

提示：combinations_with_replacement 适用于允许重复选取同一元素的场景，如多轮投票、可重复抽样等。

八、相关方法对比

九、小结

• combinations 生成不考虑顺序的组合，不允许重复选取同一元素

• combinations_with_replacement 允许同一元素在组合中多次出现

• 结果数量为组合数 C(n,r)，比排列数量小得多

• 在概率统计、特征选择等场景中应用广泛

十、练习题