python中defaultdict避免KeyError异常

python中defaultdict避免KeyError异常

在字典处理序列中，需要通过键值来进行一系列的运行。但是会出现一些不存在的键，影响程序运行抛出报错。为了避免这种情况的产生，我们可以使用defaultdict来解决。下面我们就defaultdict的概念、参数进行介绍，然后就使用defaultdict方法后，程序运行的不同情况带来实例展示。

1.概念

defaultdict是Python内建dict类的一个子类，第一个参数为default_factory属性提供初始值，默认为None。它覆盖一个方法并添加一个可写实例变量。它的其他功能与dict相同，但会为一个不存在的键提供默认值，从而避免KeyError异常。

2.参数

第一个参数是键的名称；

第二个参数是默认值。假如字典中不存在给定的键，则返回参数中提供的默认值；反之，则返回字典中保存的值。

3.KeyError异常分析

bag = ['apple', 'orange', 'cherry', 'apple','apple', 'cherry', 'blueberry']
count = {}
for fruit in bag:
    count[fruit] += 1
错误： KeyError: 'apple'

使用 collections 模块中的 defaultdict 来构造这样的字典defaultdict类避免KeyError异常，defaultdict 的一个特征是它会自动初始化每个 key 刚开始对应的值。

In [1]: l = [('a',2),('b',3),('a',1),('b',4),('a',3),('a',1),('b',3)]
In [2]: from collections import defaultdict
In [3]: d = defaultdict(list)
In [4]: for key, value in l:
   ...:     d[key].append(value)
   ...:
 
In [5]: d
Out[5]: defaultdict(list, {'a': [2, 1, 3, 1], 'b': [3, 4, 3]})

以上就是python中defaultdict避免KeyError异常的方法，从上面的代码中我们可以看到，如果没有defaultdict赋予默认值，遇到不存在键时就会产生错误了。