Python中如何对含嵌套子列表的单个列表生成笛卡尔积？

没问题，这个需求完全可以实现！你提到的itertools.product确实好用，但需要稍微调整一下思路——先把列表里的固定部分和需要展开的可变部分分开处理就行。

场景1：单个可变嵌套部分 #

针对你给出的示例，我们可以直接定位到那个包含多个子元素的嵌套项，然后遍历它的每个元素，和前后固定部分拼接成新列表：

original = [['a'], ['b'], [[1], [2], [3]], ['d']]

# 拆分固定前缀、可变部分、固定后缀
fixed_prefix = original[:2]  # 取前两个固定元素
var_section = original[2]    # 需要展开的嵌套列表
fixed_suffix = original[3:]  # 取后面的固定元素

# 生成所有结果组合
results = [fixed_prefix + [item] + fixed_suffix for item in var_section]

# 输出结果
for res in results:
    print(res)

运行后会得到你想要的输出：

[['a'], ['b'], [1], ['d']]
[['a'], ['b'], [2], ['d']]
[['a'], ['b'], [3], ['d']]

场景2：多个可变嵌套部分 #

如果你的列表里有多个需要展开的嵌套项，这时候itertools.product就能派上用场了——我们可以先提取所有可变部分，生成它们的笛卡尔积，再替换回原列表的对应位置：

import itertools

original = [['a'], [[1], [2]], ['b'], [[3], [4]]]

# 标记所有需要展开的项的索引
var_indices = {1, 3}
# 构建模板：用None作为可变部分的占位符，保留固定部分
template = []
var_parts = []
for idx, elem in enumerate(original):
    if idx in var_indices:
        template.append(None)
        var_parts.append(elem)
    else:
        template.append(elem)

# 生成所有可变部分的笛卡尔积，替换模板中的占位符
for combo in itertools.product(*var_parts):
    combo_iter = iter(combo)
    result = [next(combo_iter) if elem is None else elem for elem in template]
    print(result)

这段代码会输出所有可能的组合：

[['a'], [1], ['b'], [3]]
[['a'], [1], ['b'], [4]]
[['a'], [2], ['b'], [3]]
[['a'], [2], ['b'], [4]]

核心思路总结 #

• 先区分列表中的固定元素和需要展开的可变嵌套元素
• 单个可变元素：直接遍历拼接即可，简单高效
• 多个可变元素：用itertools.product生成所有可能的组合，再替换回模板对应的位置

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