如何通过Python的itertools模块使用product函数高效生成多个列表的笛卡尔积？

本文共计768个文字，预计阅读时间需要4分钟。

pythonimport itertools

itertools.product 生成多个可迭代对象的笛卡尔积返回一个生成器，每个元素是元组形式的组合示例：

• 最简用法：list(product([1, 2], ['a', 'b'])) → [(1, 'a'), (1, 'b'), (2, 'a'), (2, 'b')]
• 多个序列支持任意数量参数：product([1, 2], ['x'], [True, False])
• 单个序列重复使用需显式传入 repeat 参数：product('ab', repeat=3) 等价于 product('ab', 'ab', 'ab')

常见错误：把列表当单个参数传入导致嵌套

写成 product([[1, 2], ['a', 'b']]) 会把整个列表当作一个可迭代对象，结果只有一项：([[1, 2], ['a', 'b']],) —— 这不是笛卡尔积，是“包了一层”。

• 正确解包写法：product(*[[1, 2], ['a', 'b']])
• 动态参数场景（比如从变量 lists = [[1,2], ['x','y'], [True]] 构造）必须加 *，否则逻辑全错
• 如果误用 product(lists)，输出长度永远是 1，容易被忽略，建议先用 next() 检查首项结构

与嵌套 for 循环和列表推导式的性能与语义差异

product 本质是延迟计算的迭代器，不等价于 [(i,j) for i in a for j in b] 这种立即求值的列表推导式。

• 内存：product 不构建中间列表，适合大集合（如 product(range(1000), range(1000))）
• 行为：product(a, b) 中若 a 或 b 是生成器，它会在每次 next() 时重新遍历 —— 注意这不是 bug，而是设计：每个外层元素配齐所有内层元素，所以内层必须可重放
• 若内层是耗尽型生成器（如 (x for x in range(3))），需先转为 tuple 或 list，否则第二轮外层迭代时内层为空

处理非均匀长度或含空序列的边界情况

空序列参与笛卡尔积时，结果直接为空 —— 这和数学定义一致，但容易在配置驱动的代码中引发静默失败。

立即学习“Python免费学习笔记（深入）”；

• product([1,2], []) 返回空迭代器，list(...) 得 []，不会报错
• 如果业务上要求至少一个组合，需提前校验：if not all(seq for seq in sequences): raise ValueError("Empty sequence not allowed")
• 含 None 或不可迭代对象会抛 TypeError: 'NoneType' object is not iterable，product 不做类型宽容

实际用的时候，最常踩的坑就两个：忘记解包 *，以及没意识到空序列会让整个结果消失。这两点不验证，调试时可能花半天才定位到源头。