Python中的defaultdict函数与Trie树

最新推荐文章于 2023-01-01 21:15:59 发布

原创最新推荐文章于 2023-01-01 21:15:59 发布 · 697 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#自然语言处理 #pytorch

收录于

NLP 专栏收录该内容

27 篇文章

订阅专栏

博客介绍了Python中普通dict和defaultdict的区别。普通dict在key不存在时会报错，而defaultdict可在key不存在时返回默认值。默认值通过给defaultdict()传参设定，传入函数，key不存在时返回该函数默认返回值，如int函数默认值为0，也可用自定义函数。

dict()

对于普通的dict，key不存在就报错。

>>> a = dict()
>>> a['k1'] = 1
>>> a
{'k1': 1}
>>> a['k2']
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
KeyError: 'k2'
>>>

from collections import defaultdict

用于产生一个带有默认值的dict。主要针对key不存在的情况下，也希望有返回值的情况。

但是对于defaultdict，key不存在就会返回默认值。

那么默认值如何设定？通过给defaultdict()方法传参，传入的是一个函数（或叫做方法），当key不存在时，返回的就是这个函数的默认返回值。

>>> a = defaultdict(int)
>>> a
defaultdict(<class 'int'>, {})
>>> a['k2']
0
>>>

此时就不会报错。而且返回值是0。

我们还发现了，原来int这个内置函数的默认值是0。看这个示例：

>>> print(int())
0

这里的函数可以是任意自定义函数，只要又default的return即可。

>>> from collections import defaultdict
>>> def fun(): return 666
...
>>> fun()
666
>>> a = defaultdict(fun)
>>> a
defaultdict(<function fun at 0x10a8185f0>, {})
>>> a['nonexist_key']
666
>>>