如何以Pythonic方式生成连续递增列表的元素组合？

问题场景 #

给定若干个列表，每个列表ln包含从0到n-1的整数，示例如下：

l1 = [0]
l2 = [0, 1]
l3 = [0, 1, 2]

需要生成这些列表的元素笛卡尔积，即所有可能的元素组合，输出结果如下：

0 0 0 
0 0 1 
0 0 2 
0 1 0 
0 1 1
0 1 2

符合Python风格的高效实现 #

Python标准库中的itertools.product是解决这类问题的最优方案——它由底层C实现，效率极高，同时代码简洁直观，完全符合Python的设计风格。

基础用法

直接传入所有目标列表，遍历生成的组合即可：

import itertools

l1 = [0]
l2 = [0, 1]
l3 = [0, 1, 2]

# 遍历所有组合并格式化输出
for combo in itertools.product(l1, l2, l3):
    print(' '.join(map(str, combo)))

动态列表集合的处理

如果输入的列表数量不固定，可以将所有列表存入一个容器，用*解包后传入product，适配任意数量的输入列表：

import itertools

lists = [[0], [0, 1], [0, 1, 2]]

for combo in itertools.product(*lists):
    print(' '.join(map(str, combo)))

为什么这是最优解 #

• 效率高：itertools模块的核心逻辑由C编写，比手动实现嵌套循环或递归的Python代码运行速度快得多；
• 代码简洁：一行核心代码即可生成所有组合，可读性强，符合Python"简洁胜于复杂"的设计哲学；
• 灵活性强：支持任意数量的输入列表，无需修改核心逻辑即可适配不同场景。

内容的提问来源于stack exchange，提问作者HaakonA