Python中如何定义函数实现字典过滤并为缺失指定键设置默认值
Python中如何定义函数实现字典过滤并为缺失指定键设置默认值
这个需求其实挺常见的,我给你分享几种简洁又实用的实现方式,保证能完美满足你的要求:
方法一:字典推导式(最简洁)
直接通过遍历目标默认字典的键,用dict.get()方法来获取值——如果原始字典里有这个键就用它的值,没有就用默认值,一行代码就能搞定:
def f(paramsToKeepWithDefaults, params): return {key: params.get(key, paramsToKeepWithDefaults[key]) for key in paramsToKeepWithDefaults}
咱们用你给的例子测试一下:
params = {'a':1,'b':2,'z':'bad'} paramsToKeepWithDefaults = {'a':0, 'c':0} assert(f(paramsToKeepWithDefaults, params) == {'a':1,'c':0})
完全符合预期,运行不会报错~
方法二:先复制默认值再更新(逻辑更直观)
如果觉得字典推导式有点抽象,也可以用“先复制默认字典,再用原始字典中存在的目标键值覆盖”的思路,逻辑更清晰,新手也能一眼看懂:
def f(paramsToKeepWithDefaults, params): # 先复制一份默认值字典作为结果基础 result = paramsToKeepWithDefaults.copy() # 只筛选原始字典中存在于目标键集合里的键值对,更新到结果中 valid_params = {k: v for k, v in params.items() if k in paramsToKeepWithDefaults} result.update(valid_params) return result
这个方法的好处是分步明确,先把所有默认值都放好,再用合法的参数去覆盖,同样能得到你想要的结果。
补充说明
两种方法的核心逻辑都是只保留目标字典里的键,同时优先使用原始字典中的值,缺失时才用默认值。你可以根据自己的代码风格选择合适的实现方式~
备注:内容来源于stack exchange,提问作者phhu
相关产品推荐
相关产品推荐

