通八洲科技

生成所有嵌套列表的笛卡尔积组合(无需嵌套循环)

日期:2025-12-26 00:00 / 作者:心靈之曲

使用 python 标准库 `itertools.product` 可高效生成任意数量子列表的笛卡尔积,避免硬编码嵌套循环,支持动态长度输入。

在处理多维组合问题时,例如从多个列表中各取一个元素构成所有可能的元组(即笛卡尔积),若列表数量 n 在运行时才确定,传统 n 层 for 循环将无法实现。此时,itertools.product 是最简洁、高效且符合 Python 风格的解决方案。

itertools.product 接收多个可迭代对象,并返回其笛卡尔积的迭代器。配合解包操作符 *,可直接将“列表的列表”展开为独立参数:

import itertools

# 示例输入:任意长度的列表嵌套(每个子列表可含不同元素)
data = [
    ['a', 'b', 'c'],
    ['p', 'q', 'r'],
    ['x', 'y', 'z']
]

# 生成所有组合(返回元组形式)
combinations = list(itertools.product(*data))

# 若需列表而非元组(如题目示例中为 list of lists)
combinations_as_lists = [list(combo) for combo in combinations]

print(combinations_as_lists[:6])  # 查看前6个结果
# 输出:
# [['a', 'p', 'x'], ['a', 'p', 'y'], ['a', 'p', 'z'],
#  ['a', 'q', 'x'], ['a', 'q', 'y'], ['a', 'q', 'z']]

关键优势

⚠️ 注意事项

总之,itertools.product(*lists) 是解决“未知层数的组合枚举”问题的标准答案——简洁、健壮、高性能,是 Python 工程实践中推荐的惯用写法。