如何修改Python嵌套字典键存在性检查代码以获取目标键的层级路径

你的需求是既要检查目标键是否存在，还要返回它的层级路径（比如spam.egg），原来的函数只返回布尔值，我们可以修改递归逻辑，让它跟踪并返回路径，而不是单纯的存在性结果。

单路径查找（返回第一个匹配的路径） #

下面是修改后的实现，它会返回第一个找到的目标键的完整路径，找不到则返回None：

data = { "spam": { "egg": { "bacon": "Well..", "sausages": "Spam egg sausages and spam", "spam": "does not have much spam in it", } } }

def find_nested_key_path(dictionary, target_key, current_path=""):
    # 检查当前字典是否包含目标键
    if target_key in dictionary:
        # 拼接路径：如果当前路径为空，直接返回键名；否则用点分隔拼接
        return current_path + target_key if current_path else target_key
    
    # 遍历所有子字典，递归查找
    for key, value in dictionary.items():
        if isinstance(value, dict):
            # 递归调用时，把当前键加入路径（末尾加点，方便后续拼接）
            nested_path = find_nested_key_path(value, target_key, current_path + key + ".")
            # 如果找到有效路径，直接返回，不再继续遍历
            if nested_path is not None:
                return nested_path
    
    # 遍历完所有可能都没找到，返回None
    return None

# 测试查找"bacon"
result_path = find_nested_key_path(data, "bacon")
if result_path:
    print(f"找到目标键的路径：{result_path}")
    # 示例：根据路径获取对应的值（后续迭代操作可以基于此）
    current_data = data
    for path_part in result_path.split("."):
        current_data = current_data[path_part]
    print(f"对应的值为：{current_data}")
else:
    print("未找到目标键")

逻辑说明 #

• 新增current_path参数，用来跟踪递归过程中已经走过的路径
• 每次进入子字典时，把当前键拼接到路径末尾（加.作为分隔符）
• 一旦找到目标键，就返回完整路径；如果递归找到路径，直接向上传递结果
• 找不到时返回None，方便后续判断

多路径查找（返回所有匹配的路径） #

如果你的字典里可能存在多个同名的目标键，想要获取所有路径，可以用下面的版本：

def find_all_nested_key_paths(dictionary, target_key, current_path="", paths=None):
    # 初始化路径列表（避免递归时共享同一个列表的问题）
    if paths is None:
        paths = []
    
    # 如果当前字典有目标键，记录路径
    if target_key in dictionary:
        full_path = current_path + target_key if current_path else target_key
        paths.append(full_path)
    
    # 遍历子字典，递归收集所有路径
    for key, value in dictionary.items():
        if isinstance(value, dict):
            find_all_nested_key_paths(value, target_key, current_path + key + ".", paths)
    
    return paths

# 测试查找所有名为"spam"的键的路径
all_paths = find_all_nested_key_paths(data, "spam")
print("所有匹配的路径：", all_paths)

这个版本会返回一个包含所有匹配路径的列表，比如上面的测试会输出['spam', 'spam.egg.spam']。

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