如何用collections.defaultdict实现多键优先查找并返回空值

嘿，我来帮你搞定这个需求！结合defaultdict的特性，我们可以优雅地实现“先找键'A'，再找键'B'，都没有返回空字符串”的逻辑，还能彻底避免KeyError。下面给你两种实用的实现方式：

方法一：快速适配原字典的直接查找 #

如果你只是想针对现有字典做一次性的优先级查找，可以先把普通字典包装成defaultdict，再按顺序取值：

from collections import defaultdict

# 你的原字典
X = {'A': '1', 'B': '2', 'C': '3'}

# 转换为defaultdict，指定默认值为空字符串
default_X = defaultdict(lambda: '', X)

# 按优先级获取值：先A，再B，都没有就用默认空串
Y = default_X['A'] if default_X['A'] else default_X['B']

这种方式简单直接，把原字典的键值对导入defaultdict后，任何不存在的键都会返回空字符串，不用再手动判断键是否存在。

方法二：封装成通用函数（更推荐） #

如果你需要复用这个“传入两个键按顺序查找”的逻辑，把它封装成函数会更方便：

from collections import defaultdict

def priority_lookup(target_dict, first_key, second_key):
    # 将目标字典转为defaultdict，默认返回空字符串
    dd = defaultdict(lambda: '', target_dict)
    # 利用or逻辑实现优先级：先取第一个键的值，为空则取第二个
    return dd[first_key] or dd[second_key]

# 测试几个场景看看效果
test_dict = {'A': '1', 'B': '2', 'C': '3'}
print(priority_lookup(test_dict, 'A', 'B'))  # 输出: '1'（找到A）
print(priority_lookup(test_dict, 'D', 'B'))  # 输出: '2'（没找到A，找到B）
print(priority_lookup(test_dict, 'D', 'E'))  # 输出: ''（A和B都没找到）

关键逻辑说明： #

• defaultdict(lambda: '', target_dict)：这一步把普通字典转换成defaultdict，同时保留原有的键值对；指定的lambda函数会在访问不存在的键时返回空字符串，从根源上避免了KeyError。
• dd[first_key] or dd[second_key]：Python中or会返回第一个为真的值，所以如果first_key存在且对应值非空，就返回它；否则返回second_key的值——如果second_key也不存在，就会触发defaultdict的默认行为，返回空字符串，完全符合你的要求。

另外提一句，你原代码里用"A" in str(X)来判断键是否存在是多余的，直接用"A" in X就可以准确判断，但用defaultdict之后连判断步骤都省啦！

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