Python如何避免多层嵌套循环 高效校验多列表元素组合条件

你要的无硬编码嵌套、适配任意n个列表的高效实现，可以直接用标准库完成，不需要自己写复杂的递归逻辑。

基础通用版本（和原代码逻辑完全一致） #

用itertools.product生成所有输入列表的笛卡尔积，这个方法是C层实现的，性能和你手写三层for循环几乎没有差别，不管传多少个列表进去都能正常工作：

from itertools import product

# 把所有需要遍历的列表都放到这个大列表里，后续增减列表只需要改这里
all_lists = [
    [4, 5, 3],
    [1, 2, 3, 4],
    [1, 0]
]
LIMIT = 9

for combo in product(*all_lists):
    if sum(combo) < LIMIT:
        print(*combo)

运行结果和你给出的三层嵌套代码完全一致。

版本特点 #

• 无硬编码嵌套：不管是3个列表还是10个列表，只要把列表加到all_lists中即可，不需要修改循环逻辑
• 性能足够：日常场景下这个版本的速度和手写嵌套循环持平，没有额外的Python层开销
• 代码简洁：核心逻辑只有几行，维护成本极低

大数据量优化版本（剪枝优化） #

如果你的列表数量多、单个列表元素多，全量生成笛卡尔积会出现组合爆炸的问题，这时候可以加提前剪枝逻辑——当前缀元素的和已经大于等于阈值时，直接跳过这个前缀对应的所有后续组合，不需要等所有元素选完再判断，能大幅减少无效计算：

def get_valid_combos(lists, limit, current_sum=0, current_prefix=()):
    if not lists:
        if current_sum < limit:
            yield current_prefix
        return
    curr_list, *remain_lists = lists
    for num in curr_list:
        new_sum = current_sum + num
        # 前缀和已经超限，直接剪枝，跳过这个分支下所有组合
        if new_sum >= limit:
            continue
        yield from get_valid_combos(remain_lists, limit, new_sum, (*current_prefix, num))

# 调用示例
all_lists = [
    [4, 5, 3],
    [1, 2, 3, 4],
    [1, 0]
]
for combo in get_valid_combos(all_lists, 9):
    print(*combo)

这个版本在列表规模大的时候，性能比全量遍历笛卡尔积高几个量级。

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