返回顶部
首页 > 资讯 > 后端开发 > Python >Python defaultdict教程示例详解
  • 591
分享到

Python defaultdict教程示例详解

Python defaultdict教程Python defaultdict 2023-01-11 18:01:36 591人浏览 安东尼

Python 官方文档:入门教程 => 点击学习

摘要

1. 概述 在我们使用python中的dict时,常常需要判断某个关键字是否已经在dict中,如果不存在则初始化,存在则进行另外的操作。例如统计一篇文章中所有单词出现次数的代码,大

1. 概述

在我们使用python中的dict时,常常需要判断某个关键字是否已经在dict中,如果不存在则初始化,存在则进行另外的操作。例如统计一篇文章中所有单词出现次数的代码,大致写法如下:

Words_num = {}
for word in words:
    if word not in words_num.keys():
        words_num[word] = 1
    else:
        words_num[word] += 1

这样写总是需要判断key是否在dict中,不是很优雅。

Python标准库collections中defaultdict类可以很好的解决这个问题。这个类使用与dict几乎一样,除了可以在初始化时设置key的默认类型和数值。类的声明格式为defaultdict(default_factory=None, /[, ...]),default_factory是一个Callable类型的变量。

例如,foo = defaultdict(int)表示foo中的key的默认类型是int,且默认值为int的默认值0,我们可以获取任意的key,不需要手动初始化key:

>>> from collections import defaultdict
>>> foo = defaultdict(int)
>>> foo['a']
0
>>> foo['b']
0
>>> foo['whatever']
0
>>> foo['a'] += 1
>>> foo['a']
1

所以最开始的例子可以简化为如下:

from collections import defaultdict
words_num = defaultdict(int)
for word in words:
    words_num[word] += 1

可以看到使用defaultdict后,代码中只需要关注上层逻辑(统计单词的出现次数),而不需要关注具体的语法的代码实现(dict是否存在某个key,没有的话xxx,有的话xxx),因此世界变得更美好了一些。

除了int外,用list,tuple,dict,set等作为变量也比较常见。除了内置类型外,还可以自定义函数,比如设置key的默认值为'China':

>>> from collections import defaultdict
>>> def set_default_contry():
...     return "China"
...
>>> person_from = defaultdict(set_default_contry)
>>> person_from['张三']
'China'
>>> person_from['李四']
'China'
>>> person_from['Tim'] = 'USA'
>>> person_from
defaultdict(<function set_default_contry at 0x10896eca0>, {'a': 'China', '张三': 'China', '李四': 'China', 'Tim': 'USA'})

defauldict是一个简单但很好用的功能,在日常的使用中还是能减少一些代码复杂度的。希望这篇小文能给让你写代码更容易,更开心。

以上就是Python defaultdict教程示例详解的详细内容,更多关于Python defaultdict教程的资料请关注编程网其它相关文章!

--结束END--

本文标题: Python defaultdict教程示例详解

本文链接: https://lsjlt.com/news/177429.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com    QQ/279061341

猜你喜欢
软考高级职称资格查询
编程网,编程工程师的家园,是目前国内优秀的开源技术社区之一,形成了由开源软件库、代码分享、资讯、协作翻译、讨论区和博客等几大频道内容,为IT开发者提供了一个发现、使用、并交流开源技术的平台。
  • 官方手机版

  • 微信公众号

  • 商务合作