Python Academy · 课时

itertools：组合数学

Q: 「itertools：组合数学」这节课中我会学到什么？

生成排列、组合和笛卡尔积。 你通过在浏览器中直接运行的动手代码来练习 Python Academy，全天候 AI 导师会在你学习这节课的过程中回答你的问题。

生成排列、组合和笛卡尔积。

itertools：组合数学 是 CoddyKit 上的免费 Python Academy 课时。这是第 2 节课，共 4 节。你可以在下方免费阅读本课时的完整内容 — 然后在浏览器中使用内置代码编辑器和全天候 AI 导师进行实践。这是 Python Academy 学习路径的一部分，你的进度在网页和 CoddyKit 应用中同步。 Python Academy 课程共包含 4 节课。

组合数学概览

itertools 提供四个组合数学函数：product、permutations、combinations 和 combinations_with_replacement。

import itertools

# All orderings of 2 items from ABC
print(list(itertools.permutations("ABC", 2)))
# [(A,B),(A,C),(B,A),(B,C),(C,A),(C,B)]

product()

product(*iterables, repeat=1) 会计算笛卡尔积，效果等同于嵌套的 for 循环。

import itertools

print(list(itertools.product([1,2], ["a","b"])))
# [(1,"a"),(1,"b"),(2,"a"),(2,"b")]

# repeat=2 pairs each element with itself
print(list(itertools.product(range(2), repeat=2)))
# [(0,0),(0,1),(1,0),(1,1)]

permutations()

permutations(it, r) 会生成所有长度为 r 的有序排列。总数为：P(n,r) = n!/(n-r)!。

import itertools

result = list(itertools.permutations([1,2,3], 2))
print(result)
# [(1,2),(1,3),(2,1),(2,3),(3,1),(3,2)]
print(len(result))  # 6

combinations()

combinations(it, r) 会生成所有长度为 r、无序且不重复的选择。总数为：C(n,r) = n!/(r!(n-r)!)。

import itertools

result = list(itertools.combinations([1,2,3,4], 2))
print(result)
# [(1,2),(1,3),(1,4),(2,3),(2,4),(3,4)]
print(len(result))  # 6

combinations_with_replacement()

combinations_with_replacement(it, r) 允许同一个元素在一个组合中出现多次。

import itertools

result = list(itertools.combinations_with_replacement("AB", 2))
print(result)
# [(A,A),(A,B),(B,B)]

无需生成全部元素即可计数

请使用 math.perm、math.comb 或 len() 快捷方式（适用于有限结果），而不是仅仅为了计数就生成所有元素。

import math

print(math.perm(10, 3))   # 720
print(math.comb(10, 3))   # 120

生成密码或密钥

组合数学迭代器适合生成候选密钥或测试用例，无需将全部内容加载到内存中。

import itertools, string

chars = string.ascii_lowercase
# All 2-char lowercase combos:
for combo in itertools.combinations(chars, 2):
    pass  # process without materialising

使用 product 生成网格坐标

使用 product(range(rows), range(cols)) 遍历二维网格，无需编写嵌套循环。

import itertools

for row, col in itertools.product(range(3), range(3)):
    print(f"({row},{col})", end=" ")

测试所有子集

对于从 0 到 n 的每个长度，分别遍历 combinations，即可生成列表的所有子集。

import itertools

items = [1, 2, 3]
all_subsets = []
for r in range(len(items)+1):
    all_subsets.extend(itertools.combinations(items, r))
print(all_subsets)

使用 combinations 去重

使用 combinations 恰好比较每一对元素一次，从而避免重复比较 (a,b) 和 (b,a)。

import itertools

words = ["apple","apricot","banana","blueberry"]
for a, b in itertools.combinations(words, 2):
    if a[0] == b[0]:
        print(f"Same letter: {a}, {b}")

性能注意事项

组合数学序列的增长速度非常快。permutations(range(12)) 会生成 4.79 亿个结果。请始终使用生成器，并且只将所需内容生成到内存中。

import itertools, math

n = 12
print(f"P(12,12) = {math.factorial(n):,}")   # 479,001,600
# Never: list(itertools.permutations(range(12)))
# Instead: iterate lazily and break early

快速检查

哪个 itertools 函数可以从一个集合中生成所有无序且不重复的元素对？

复习

使用 product 计算笛卡尔积，使用 permutations 生成有序排列，使用 combinations 生成无序子集，并在元素可以重复时使用 combinations_with_replacement。请始终以惰性方式处理组合数学迭代器。

常见问题解答

「itertools：组合数学」课时是免费的吗？

是的 — 「itertools：组合数学」的完整文本可在网页上免费阅读。要进行交互式练习（内置代码编辑器和全天候 AI 导师）并解锁 Python Academy 课程的其余内容，请升级到 CoddyKit PRO。 Python Academy 课程共包含 4 节课。

「itertools：组合数学」这节课中我会学到什么？

生成排列、组合和笛卡尔积。你通过在浏览器中直接运行的动手代码来练习 Python Academy，全天候 AI 导师会在你学习这节课的过程中回答你的问题。

学习 Python Academy 需要有经验吗？

无需任何先前经验。CoddyKit 上的 Python Academy 课程适合初学者到高级学习者，你可以从这里开始或从头开始，按照自己的节奏学习。这是第 2 节课，共 4 节。

「itertools：组合数学」课时需要多长时间？

大多数 CoddyKit 课程大约需要 5–10 分钟。每节课都很精短且互动，所以你能稳步进步，并在网页和应用中从离开的地方继续。

我能在这节 Python Academy 课中编写并运行代码吗？

能。每节 Python Academy 课都包含内置代码编辑器，你可以在浏览器中直接编写并运行真实代码，并获得即时 AI 反馈 — 无需本地设置。

此课程中的所有课时

← 返回 Python Academy