Python中基于嵌套字典值查找顶层键的代码优化求助

你的原代码存在两处可优化的点：一是频繁将字典的values()转换为列表，增加了不必要的开销；二是通过索引匹配根键的写法不够直观。结合所有子值唯一的前提，这里提供两种更高效简洁的实现方案：

方案一：预构建反向映射（高频查询首选） #

因为所有子字典的值都是唯一的，我们可以提前构建一个值到根键的反向字典，后续查询直接通过键值对获取，时间复杂度为O(1)，适合需要多次调用查询函数的场景。

maindict = {"First Node"  : {0: "00", 1: "01", 2: "02", 3: "03"},
        "Second Node" : {0: "10", 1: "11", 2: "12", 3: "13"},
        "Third Node"  : {0: "20", 1: "21", 2: "22", 3: "33"}}

# 预构建反向映射字典
reverse_map = {v: root_key for root_key, sub_dict in maindict.items() for _, v in sub_dict.items()}

def search_root(val):
    # 找不到时返回None，可根据需求调整默认值
    return reverse_map.get(val, None)

方案二：简化单次查询的循环逻辑（低频查询首选） #

如果只是偶尔查询，不需要预构建反向字典，可以直接遍历原字典的键值对，省去列表转换和索引匹配的冗余操作，代码更简洁直观：

maindict = {"First Node"  : {0: "00", 1: "01", 2: "02", 3: "03"},
        "Second Node" : {0: "10", 1: "11", 2: "12", 3: "13"},
        "Third Node"  : {0: "20", 1: "21", 2: "22", 3: "33"}}

def search_root(val):
    for root_key, sub_dict in maindict.items():
        if val in sub_dict.values():
            return root_key
    # 未找到目标值时返回None
    return None

对比原代码的优势 #

• 避免了list(maindict.values())、list(dict.values())这类不必要的列表转换，减少内存开销
• 直接遍历键值对，无需通过索引匹配根键，逻辑更清晰
• 方案一的反向映射让查询效率从O(n)提升到O(1)，高频场景下性能提升明显

内容的提问来源于stack exchange，提问作者Ruan Carlo Weiers Britzke