Python中动态生成嵌套列表索引赋值循环的实现方法咨询

需求背景 #

给定嵌套列表：[4, [1], [5, [-2, 10], [1, -7] ], 6, [-2, [1, [0]], [9] ], [-3], [7, 3] ]，需逐层对长度大于1的子列表应用函数funct：先处理外层符合条件的子列表，再处理这些子列表内部的符合条件子列表，依此类推。

当前实现为每一层级编写独立for循环，例如：

• 第一步（处理外层）：

for i in range(len(var)):
    A,B=funct(var[i])
    aux.append(B)
    var[num[0][i]]=A

• 第二步（处理下一层）：

for i in range(len(var)):
    A,B=funct(var[i])
    aux.append(B)
    var[num[0][i][0]][num[0][i][1]]=A

每一步仅为var的索引增加[num[0][i][j]]层级，但由于无法预知嵌套列表的结构深度，这种写法冗余度极高。希望通过while循环动态添加索引层级，实现通用处理。

注：num变量记录各层级符合条件的子列表位置：

• 第一步：[2, 4, 6]
• 第二步：[[2, 1], [2, 2], [4, 2]]

通用实现方案 #

当然可以通过while循环配合索引路径遍历来实现通用处理，核心是动态遍历num中记录的索引路径，定位嵌套列表里的目标子列表，无需硬编码多层索引。

实现代码示例 #

aux = []
# 初始化为第一层需要处理的索引路径
current_level_paths = num[0]  

while current_level_paths:
    next_level_paths = []
    for path in current_level_paths:
        # 按照索引路径定位到目标子列表
        target_item = var
        for idx in path:
            target_item = target_item[idx]
        
        # 应用funct函数，获取返回值
        A, B = funct(target_item)
        aux.append(B)
        
        # 更新原嵌套列表中的对应位置
        parent_node = var
        # 遍历到父节点（路径的倒数第二个索引）
        for idx in path[:-1]:
            parent_node = parent_node[idx]
        # 替换父节点中对应索引的元素
        parent_node[path[-1]] = A
        
        # 收集当前子列表中需要处理的下一层路径（长度>1的子列表）
        for sub_idx, sub_item in enumerate(A):
            if isinstance(sub_item, list) and len(sub_item) > 1:
                next_level_paths.append(path + [sub_idx])
    
    # 切换到下一层继续处理
    current_level_paths = next_level_paths

关键细节说明 #

• 动态索引定位：通过遍历索引路径（如[2,1]），逐层深入嵌套列表，无需写死var[2][1]这种硬编码层级。
• 原列表更新逻辑：必须先找到目标子列表的父节点，再修改父节点的对应索引元素，直接修改target_item不会影响原列表（因为赋值操作会改变引用）。
• 层级迭代：每处理完一层，收集下一层需要处理的索引路径，直到没有符合条件的子列表为止，实现全深度自动处理。

内容的提问来源于stack exchange，提问作者David Fígols Olaria