Python 中的 defaultdict 数据类型

Python 中的 defaultdict 数据类型

官方网址

首先，defaultdict 是 dict 的一个子类。通常 Python 中字典（dict）这种数据类型是通过键值对来存取的，当索引一个不存在的键时，就会引发 keyerror 异常。那么，defaultdict 就可以解决这个问题，它可以实现为不存的键值返回一个默认值。

defaultdict是 collections 包下的一个模块，defaultdict 在初始化时可以提供一个 default_factory 的参数，default_factory 接收一个工厂函数作为参数， 可以是 int、str、list 等内置函数，也可以是自定义函数。

用法实例如下：

示例一： 统计字符串中字母出现的个数。

from collections import defaultdict

s = 'mississippi'
d = defaultdict(int)
for k in s:
    d[k] += 1
print(d)

# 输出
# defaultdict(<class 'int'>, {'m': 1, 'i': 4, 's': 4, 'p': 2})

如果不用 defaultdict 的话，写法如下：

s = 'mississippi'
d = {}
for k in s:
    if k not in d:
        d[k] = 1
    else:
        d[k] = d[k] + 1
print(d)

# 输出
# {'m': 1, 'i': 4, 's': 4, 'p': 2}

s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
d = defaultdict(list)
for k, v in s:
    d[k].append(v)
print(d)

# 输出
# defaultdict(<class 'list'>, {'yellow': [1, 3], 'blue': [2, 4], 'red': [1]})

示例二： 使用 lambda 函数。

q = defaultdict(lambda: [0, 0, 0])
print(q['a'])

# 输出
# [0, 0, 0]

如果本文章对您有帮助，记得在下方点赞哟！也欢迎在下方评论区留言讨论！