时序图遍历场景下如何将带动态赋值的嵌套循环改造为递归函数

实现思路 #

• 把所有时间步的fixed配置按时间顺序整理为列表fixed_list，第k个时间步对应列表的第k-1个元素
• 自定义每步业务逻辑的回调函数step_callback，接收当前时间步索引t和当前计算结果assigned两个参数，填入你原本每个循环内部的处理逻辑
• 递归函数逐时间步执行，每步完成变量赋值、业务逻辑处理后自动进入下一层递归，直到所有时间步处理完毕自动终止，无需手动新增嵌套循环

递归实现代码 #

def time_step_recursion(t, past_assigned, fixed_list, step_callback):
    # 终止条件：所有时间步处理完成
    if t >= len(fixed_list):
        return
    # 生成当前时间步的计算函数
    if t == 0:
        current_F = func()
    else:
        current_F = func2(past_assigned)
    current_fixed = fixed_list[t]
    # 遍历所有变量的取值组合
    for j in product(*[D[i] for i in D.keys()]):
        assigned = dict(zip(D.keys(), j))
        assigned = assign(assigned, current_fixed, current_F)
        # 执行当前步的自定义业务逻辑
        step_callback(t, assigned)
        # 递归进入下一个时间步
        time_step_recursion(t + 1, assigned, fixed_list, step_callback)

使用示例 #

以你给出的3个时间步场景为例，调用方式如下：

# 按时间顺序整理fixed配置
fixed_list = [
    {'B':0},  # 第1个时间步的fixed
    {'A':1},  # 第2个时间步的fixed
    {'A':1}   # 第3个时间步的fixed
]

# 自定义业务逻辑回调，这里以打印结果为例，替换为你自己的逻辑即可
def my_step_callback(t, assigned):
    # 注意：如果需要存储结果，需要用assigned.copy()避免后续递归修改覆盖
    print(f"时间步{t} 计算结果: {assigned}")

# 启动递归
time_step_recursion(t=0, past_assigned=None, fixed_list=fixed_list, step_callback=my_step_callback)

如果需要处理N个时间步，只需要在fixed_list里按顺序添加对应时间步的fixed配置即可，无需修改递归函数本身。

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