Python字典右侧引用自身赋值时如何避免KeyError实现计数
解决方案
代码触发KeyError的本质是:Python原生字典访问不存在的键时不会自动创建键并赋默认值,直接读取mydict[v]会直接抛出键不存在异常。以下是几种可直接落地的实现方式,完全匹配「键不存在则初始化为1,键存在则值加1」的计数需求:
方法1:使用
dict.get()方法(无依赖,通用写法)
读取键值时主动指定键不存在时的返回默认值为0,再执行加1操作即可,不需要额外导包,适配所有Python版本:mydict = {} range_val = [str(i).zfill(2) for i in range(100)] for v in range_val: mydict[v] = mydict.get(v, 0) + 1后续替换遍历对象为
somedict.values()时不需要修改核心逻辑,直接替换循环的可迭代对象即可。方法2:使用
collections.defaultdict(简洁高性能写法)
初始化字典时指定值的默认类型为整数,不存在的键被访问时会自动创建,默认值为整数0,可以直接用+=运算符做累加:from collections import defaultdict mydict = defaultdict(int) range_val = [str(i).zfill(2) for i in range(100)] for v in range_val: mydict[v] += 1方法3:使用
collections.Counter(计数场景专用写法)
如果核心需求就是统计元素出现频次,可以直接用标准库内置的计数工具,不需要手写循环,多数据源统计时直接调用update方法即可累加:from collections import Counter range_val = [str(i).zfill(2) for i in range(100)] mydict = Counter(range_val) # 后续追加统计其他数据源时直接执行 # mydict.update(somedict.values())方法4:手动判断键存在性(逻辑直白但冗余,不推荐)
每次更新前先判断键是否存在,不存在则手动初始化值为1,存在则累加:mydict = {} range_val = [str(i).zfill(2) for i in range(100)] for v in range_val: if v not in mydict: mydict[v] = 1 else: mydict[v] += 1
不要尝试通过修改字典类的魔法方法实现自动默认值,除非明确知道改动的影响范围,上述四种标准写法已经覆盖所有日常开发场景。
内容的提问来源于stack exchange,提问作者SJPRO
相关产品推荐
相关产品推荐

