Python实现从n个列表各选0或1个元素生成所有可能列表

这个需求本质是带空选选项的笛卡尔积计算，核心逻辑很直接：

• 对每一个输入的子列表，扩展它的可选操作：既可以选择不拿任何元素，也可以选择拿列表里的任意一个元素
• 对所有子列表的可选操作做笛卡尔积，得到所有可能的选择组合
• 把每一组选择里挑中的元素按顺序拼接，就是符合要求的结果列表

Python标准库的itertools.product原生支持笛卡尔积计算，不需要手写递归逻辑，性能和稳定性都有保障。

from itertools import product

def get_all_valid_combinations(source_lists):
    # 为每个子列表生成可选项：空元组代表不选元素，单元素元组代表选中对应元素
    option_groups = []
    for sub_list in source_lists:
        options = [()] + [(val,) for val in sub_list]
        option_groups.append(options)
    
    result = []
    # 计算所有可选组的笛卡尔积，拼接每个组合的结果
    for select_combo in product(*option_groups):
        merged_list = list(sum(select_combo, ()))
        result.append(merged_list)
    return result

# 测试示例
if __name__ == "__main__":
    test_case = [[2], [2,3], [4], [1,2], [2], [1,4]]
    all_combinations = get_all_valid_combinations(test_case)
    print(f"总组合数：{len(all_combinations)}") # 计算得2*3*2*3*2*3=216，和输出一致
    # 验证举例的结果是否存在
    print([2,2,4,1,2,1] in all_combinations) # 输出True
    print([2,4,1,2] in all_combinations) # 输出True

总组合数的计算逻辑是：每个长度为n的子列表，有n+1种选择（选0个，或者选n个元素中的1个），把所有子列表的选择数相乘就是总组合数，上述测试用例总共有23232*3=216种合法组合。你举例中提到的“每个列表选最后一个元素得到的[2,3,2,2,4]”应为笔误，严格按规则选每个列表最后一个元素得到的结果是[2,3,4,2,2,4]，该结果同样存在于生成集合中。

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