Python中按有序键列表遍历嵌套字典的优化方案问询

你的需求是沿着指定路径遍历嵌套字典（其中包含列表嵌套字典的结构），原实现确实可以工作，但逻辑拆分较碎，存在冗余代码，而且遇到非预期类型时可能出现未处理的异常。这里提供几种更简洁、健壮的实现方式：

简洁宽松版（路径不存在时返回None） #

这个版本和原代码逻辑一致，但更紧凑，同时处理了更多边界情况：

def walk_dict(data, path):
    current = data
    for key in path:
        # 当前节点是字典，直接取对应key的值
        if isinstance(current, dict):
            current = current.get(key)
        # 当前节点是列表，遍历列表中的字典找到包含key的项
        elif isinstance(current, list):
            # 用next() + 生成器表达式，找到第一个含key的字典并取出对应值，找不到返回None
            current = next((d.get(key) for d in current if key in d), None)
        # 既不是字典也不是列表，无法继续遍历，直接返回None
        else:
            return None
        
        # 如果中途current变成None，提前终止循环
        if current is None:
            break
    return current

测试验证 #

用你的示例数据测试：

d = {'A': [{'B': [{'C': [{'D1':[]}, {'D2': []}]}]}]}

print(walk_dict(d, ['A','B','C','D1']))  # 输出 []
print(walk_dict(d, ['A','B','C','D2']))  # 输出 []
print(walk_dict(d, ['A','B','C','D3']))  # 输出 None

严格报错版（路径不存在时抛出异常） #

如果需要在路径无效时明确抛出错误（而不是返回None），可以用这个版本，方便调试定位问题：

def walk_dict_strict(data, path):
    current = data
    for step_idx, key in enumerate(path):
        if isinstance(current, dict):
            if key not in current:
                raise KeyError(f"Key '{key}' not found at path segment: {path[:step_idx+1]}")
            current = current[key]
        elif isinstance(current, list):
            found = next((d.get(key) for d in current if key in d), None)
            if found is None:
                raise KeyError(f"Key '{key}' not found in list at path segment: {path[:step_idx+1]}")
            current = found
        else:
            raise TypeError(f"Unexpected type {type(current)} at path segment: {path[:step_idx+1]} - expected dict or list")
    return current

为什么这个实现更优？ #

• 代码简洁：用单循环处理所有路径节点，避免了原代码中按索引分情况的冗余逻辑
• 健壮性更强：明确处理了字典、列表和其他类型的情况，减少潜在的AttributeError
• 灵活性高：提供宽松/严格两个版本，可根据业务需求选择
• 可读性更好：每一步的逻辑清晰，没有拆分到多个函数，便于维护

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