遍历嵌套列表按子列表首元素提取值生成新列表，递归代码出错如何修复？

一、现有递归代码的错误原因 #

• 缺少递归终止条件：代码没有处理输入列表为空的情况，当遍历到最后一个元素后仍尝试访问numbers[0]，会直接抛出索引越界错误。
• 未接收递归返回值：调用nested_list(numbers[0][1:numbers[0][0] + 1])后没有将返回的结果存入最终列表，该次调用的计算结果完全丢失。
• 分支逻辑缺失：没有处理子列表首元素为0的场景，也没有对剩余子列表的递归逻辑做正确承接。
• 变量覆盖错误：else分支中直接将numbers赋值为list(numbers[0])，覆盖了原输入列表，导致后续切片操作全部错误。

修正后的递归实现 #

def nested_list(numbers):
    # 递归终止条件：输入为空时直接返回空列表
    if not numbers:
        return []
    result = []
    current = numbers[0]
    if isinstance(current, list):
        extract_count = current[0]
        if extract_count > 0:
            # 提取当前子列表的对应元素
            result.extend(current[1:extract_count + 1])
        # 递归处理剩余的子列表
        result.extend(nested_list(numbers[1:]))
    return result

二、非递归实现方案 #

该需求完全可以不用递归实现，你的输入本身是固定两层的嵌套列表，直接遍历所有子列表提取元素即可，逻辑更简洁，也不存在递归栈溢出的风险：

def extract_list(nested_data):
    result = []
    for sub_list in nested_data:
        count = sub_list[0]
        if count > 0:
            result.extend(sub_list[1:count+1])
    return result

调用示例：

test_data = [[2, 1, 2, 3, 4], [0, 4, 5], [1, 8, 9]]
print(extract_list(test_data)) # 输出 [1, 2, 8]

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