如何创建递归生成迭代函数的Python函数？实现动态n层嵌套for循环

嘿，这个需求太实用了——手动写固定层数的嵌套循环不仅麻烦，还没法灵活应对用户输入的变化。我给你分享两种最常用的解决方案，你可以根据自己的场景选：

方法一：递归实现（适合需要复杂中间逻辑的场景） #

递归的思路其实很直观：每一层循环就对应一次递归调用，当层数减到0时，就执行你原本放在最内层循环里的操作。

举个具体的例子，假设每层循环都是遍历0到2，我们要生成n层嵌套循环，每次内层执行时打印当前所有循环变量的组合：

def dynamic_nested_loops(n, current_path=None):
    # 初始化当前路径（记录每层循环的变量值）
    if current_path is None:
        current_path = []
    # 递归终止：层数为0时，执行内层操作
    if n == 0:
        print(current_path)  # 这里替换成你实际要做的事情
        return
    # 遍历当前层的所有可能值，递归进入下一层
    for i in range(3):  # 可替换成任意你需要的可迭代对象
        dynamic_nested_loops(n - 1, current_path + [i])

# 调用示例：生成3层嵌套循环
dynamic_nested_loops(3)

如果每层循环的范围不一样（比如第一层range(2)，第二层range(4)），你可以把循环范围列表作为参数传进去，修改一下代码：

def dynamic_nested_loops_custom_ranges(loop_ranges, current_path=None):
    if current_path is None:
        current_path = []
    if not loop_ranges:
        print(current_path)
        return
    # 取出当前层的循环范围，递归处理剩余层
    current_range = loop_ranges[0]
    for i in current_range:
        dynamic_nested_loops_custom_ranges(loop_ranges[1:], current_path + [i])

# 调用示例：3层不同范围的循环
dynamic_nested_loops_custom_ranges([range(2), range(4), range(5)])

方法二：用itertools.product（简洁高效，适合简单场景） #

Python标准库的itertools.product可以直接生成多个可迭代对象的笛卡尔积，这正好对应n层嵌套循环的所有组合结果。这种方法代码更短，不需要手动写递归。

还是用刚才的例子：

import itertools

def dynamic_nested_loops_with_product(n):
    # 生成n个相同的循环范围（可自定义）
    loop_ranges = [range(3)] * n
    # product会直接输出所有可能的组合，相当于n层嵌套循环的结果
    for combination in itertools.product(*loop_ranges):
        print(combination)  # 替换成你的内层操作

# 调用示例：生成3层嵌套循环
dynamic_nested_loops_with_product(3)

如果每层范围不同，直接传入不同的可迭代对象就行：

for combination in itertools.product(range(2), range(4), range(5)):
    print(combination)

两种方法各有优劣：递归更灵活，能在每层循环中插入额外逻辑；itertools.product则更简洁高效，适合只需要遍历所有组合的场景。你可以根据自己的需求来选~

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