Python如何根据用户输入的数值动态生成指定层数的嵌套for循环

多层嵌套for循环的本质是对多个可迭代对象求笛卡尔积，每一轮的输出对应各层循环的变量组合，基于这个逻辑有两种常用实现方式：

方案1：使用标准库itertools.product（推荐） #

这是最简单的实现方式，itertools.product原生支持多可迭代对象的笛卡尔积计算，完全等价于多层嵌套for循环：

import itertools

# 提前把所有层要遍历的列表放到一个总列表里
all_lists = [list1, list2, list3, list4, list5] # 可提前预置足够多的备选遍历对象

n = int(input('你需要几层for循环？ '))
# 取前n个列表做笛卡尔积，等价于n层嵌套循环
for items in itertools.product(*all_lists[:n]):
    # items是按顺序存储的各层循环变量，n=3时items = (item, item2, item3)
    # 可直接按索引取变量，或解包为单独变量
    # 此处写入你原本最内层要执行的业务代码即可
    print(items)

• 优点：代码简洁，底层为C实现性能高，不需要自行处理循环逻辑
• 适用场景：每层循环的遍历对象可提前确定，不需要依赖上一层循环的变量

方案2：递归实现 #

如果每层循环的遍历对象依赖上一层的变量，无法提前确定，可以用递归函数模拟嵌套循环：

# 提前把所有层要遍历的列表放到总列表，也可以在递归逻辑里动态生成下一层的遍历对象
all_lists = [list1, list2, list3, list4, list5]
n = int(input('你需要几层for循环？ '))

def recursive_loop(current_level, current_values):
    # current_level 表示当前正在处理第几层循环
    # current_values 是之前所有层的循环变量列表
    if current_level == n:
        # 已经到最内层，执行业务逻辑
        print(current_values)
        return
    # 遍历当前层的列表，也可根据current_values动态生成当前层的遍历对象
    for item in all_lists[current_level]:
        # 把当前层的变量加入列表，进入下一层循环
        recursive_loop(current_level + 1, current_values + [item])

# 从第0层开始执行，初始变量列表为空
recursive_loop(0, [])

• 优点：灵活性高，支持每层循环逻辑自定义，下一层的遍历对象可以根据上一层的变量动态生成
• 适用场景：层与层之间有依赖关系的动态嵌套循环

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