使用python写代码时，是否经常遇到KeyError呢？

如何定义一个多层嵌套的字典呢？当我们使用普通字典时，用法一般是a=dict(),然后可以为字典添加元素，如a['key']='xxx'.之后便可以使用a['key']引用'xxx'这个值。如果引用的Key不存在，就会抛出KeyError。如下：

>>> a=dict()>>> a[1]Traceback (most recent call last):  File "", line 1, in <module>KeyError: 1

使用defaultdict替换dict，可以避免KeyError，这可以让我们专心于代码业务逻辑，而不是检查某个key是否存在了。

defaultdict是python内建模块collections提供的一个集合类。defaultdict会在对应key不存在时返回一个默认值，除此之外与dict没有区别。

如何导入defaultdict？

要使用defaultdict，需要先导入defaultdict，如下：

from collections import defaultdict

defaultdict可接受参数有哪些？

defaultdict可接受的参数可以分为三类，先总结一下，分别为：

1

接受None作为参数。

2

接受一个‘类型’作为参数，比如int、str、list、dict等。传入类型的默认值作为返回值。

3

接受任何不带参数的可调用函数作为参数，该函数的返回结果作为默认值。

三种类型具体分析如下：01接受None作为参数

先看一段代码，示例如下：

>>> from collections import defaultdict>>> my_dict=defaultdict(None)>>> my_dict=defaultdict()

上面两种写法是一样的，传入None和不传入参数都是指没有返回值。既然没有返回值，那么此时"my_dict=defaultdict()"与"my_dict=dict()",就完全等价了。当key不存在时，也会异常：

>>> from collections import defaultdict>>> my_dict=defaultdict(None)>>> my_dict=defaultdict()>>> my_dict[1]Traceback (most recent call last):  File "", line 1, in <module>KeyError: 1

如果只是这种用法，那么一般不使用defaultdict，直接使用dict即可。

02接受一个‘类型’作为参数

接受一个‘类型’作为参数，比如int、str、list、dict等。传入类型的默认值作为返回值。比如int、str、list、dict默认值分别为0、''、[]、{}，那么下面的代码分别返回对应值：

>>> from collections import defaultdict>>> my_dict=defaultdict(int)>>> my_dict[1]0>>> my_dict=defaultdict(str)>>> my_dict[1]''>>> my_dict=defaultdict(list)>>> my_dict[1][]>>> my_dict=defaultdict(dict)>>> my_dict[1]{}

int、str、list作为参数没有什么好说的，dict作为参数可以引申一下。该用法可以为多层嵌套结构定义默认值，理论上是可以嵌套无数层字典结构的，不过有个小坑。先看示例：

>>> from collections import defaultdict>>> my_dict=defaultdict(dict)>>> my_dict[1]{}>>> my_dict[1][2]Traceback (most recent call last):  File "", line 1, in <module>KeyError: 2

分析一下：my_dict=defaultdict(dict)使my_dict变为了一个有返回值的字典，即my_dict的每个key，不管存在与否都返回一个{}。但是这个返回的{}只是一个普通字典啊，所以当我们使用my_dict[1][2]时就抛异常了，因为my_dict[1]=dict()。

那么如何使my_dict[1]也变为一个具有返回值的字典呢？my_dict=defaultdict(defaultdict)？并没有这么简单，看示例：

>>> from collections import defaultdict>>>>>> my_dict=defaultdict(defaultdict)>>> my_dict[1]defaultdict(None, {})>>> my_dict[1][2]Traceback (most recent call last):  File "", line 1, in <module>KeyError: 2

怎么回事？为何报错了？懵...

分析一下，可以看到上面的代码里my_dict[1]输出的是'defaultdict(None, {})',先看一下下面的代码：

>>> from collections import defaultdict>>> my_dict=defaultdict()>>> my_dictdefaultdict(None, {})

输出的也是'defaultdict(None, {})'，这怎么回事啊？

我们开始说过'my_dict=defaultdict()'和普通字典是一样的，那么'my_dict=defaultdict(defaultdict)'中的'my_dict[1]'也是普通字典咯...

为什么会这样？

因为defaultdict不支持这样的嵌套传参！

那又怎么定义更多层的具有返回值的字典呢？别急，是有方法的，使用匿名函数lambda可以实现，如下：

二层嵌套

>>> from collections import defaultdict>>> my_dict=defaultdict(lambda: defaultdict(dict))>>> my_dict[1]defaultdict(<class 'dict'>, {})>>> my_dict[1][2]{}# 第三层才会异常>>> my_dict[1][2][3]Traceback (most recent call last):  File "<stdin>", line 1, in <module>KeyError: 3

三层嵌套

>>> from collections import defaultdict>>> my_dict=defaultdict(lambda: defaultdict(lambda: defaultdict(dict)))>>> my_dict[1]defaultdict(<function <lambda>.<locals>.<lambda> at 0x00000243B13A2318>, {})>>> my_dict[1][2]defaultdict(<class 'dict'>, {})>>> my_dict[1][2][3]{}# 第四层异常>>> my_dict[1][2][3][4]Traceback (most recent call last):  File "<stdin>", line 1, in <module>KeyError: 4

五层嵌套..

六层嵌套..

由此可见无数层嵌套结构都是可以的...

当然，最后一层也不一定是dict了，int、str...所有defaultdict可接受参数都行的。

由上面这种利用lambda无线嵌套的用法，我们可以引出defaultdict的第三类可接受参数了，接着看...

03任何不带参数的可调用函数作为参数

接受任何不带参数的可调用函数作为参数，该函数的返回结果作为默认值。

直接看代码：

>>> from collections import defaultdict>>> def my_return():...     return 'my_value' # 也可以返回自定义结构,如另一个嵌套字典......>>> my_dict=defaultdict(my_return)>>> my_dict[1]'my_value'>>># 多层嵌套>>> my_dict=defaultdict(lambda: defaultdict(lambda: defaultdict(lambda: defaultdict(my_return))))>>> my_dict[1][2][3][4]'my_value'

可以看到，因为这个返回的默认值，我们可以利用defaultdict做很多事情了。各位，还有什么其他好玩的用法，欢迎留言分享！

客官，喜欢就关注一下吧