温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

python的setdefault与defaultdict函数怎么用

发布时间:2022-01-17 15:21:43 阅读:184 作者:iii 栏目:大数据
Python开发者专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

这篇文章主要介绍“python的setdefault与defaultdict函数怎么用”的相关知识,小编通过实际案例向大家展示操作过程,操作方法简单快捷,实用性强,希望这篇“python的setdefault与defaultdict函数怎么用”文章能帮助大家解决问题。

有这样的需求,一个列表,里面是(key, value) 这样的键值对元组,要将它转换成一个字典对象,并将key相同的value作为一组。看代码:

data = [("p", 1), ("p", 2), ("p", 3),        ("h", 1), ("h", 2), ("h", 3)]要转换成result = {'p': [1, 2, 3], 'h': [1, 2, 3]}

python的setdefault与defaultdict函数怎么用

下面这个方法是大家都能想到的,先判断result中有没有key,没有则为其初始化一个列表,有则直接将value值append到列表中。但这段代码在Python中不怎么优雅

result = {}for (key, value) in data:    if key in result:        result[key].append(value)    else:        result[key] = [value]

setdefault

更优雅的方式就是使用setdefault方法,它是字典对象的一个实例方法,接收两个参数,用法和字典的get方法类似,但是比 get 更强大。 它可以为给字典的key设定一个默认值(如果key不在字典中的时候)

定义

 def setdefault(self, k, d=None):      """ D.setdefault(k[,d]) -> D.get(k,d),           also set D[k]=d if k not in D       """    value = D.get(k,d)    if k not in D:        D[k]=d    return value

当然,内部具体实现肯定比上面的代码要高效。二者的区别是:L get 方法设置的默认值不会改变原字典, 而setdefault设置的默认值会改变原字典的值。

>>> d = {"x":3}>>> y = d.get("y", 4)>>> y4>>> d{'x': 3}

对比

>>> y = d.setdefault("y", 4)>>> y4>>> d{'y': 4, 'x': 3}

所以,前面的需求就有了这种更优雅的写法:

result = {}data = [("p", 1), ("p", 2), ("p", 3),        ("h", 1), ("h", 2), ("h", 3)]for (key, value) in data:    result.setdefault(key, []).append(value)

defaultdict

defaultdict是属于collections 模块下的一个工厂函数,用于构建字典对象,接收一个函数(可调用)对象为作为参数。参数返回的类型是什么,key对应value就是什么类型。

>>> result = defaultdict(list)>>> resultdefaultdict(<type 'list'>, {})>>> result['a'][]

参数为 list,它就会构建一个默认value为list的字典,例如result[‘a’]的值默认就是list对象。

因此,前面这段代码可以改为:

from collections import defaultdictresult = defaultdict(list)data = [("p", 1), ("p", 2), ("p", 3),        ("h", 1), ("h", 2), ("h", 3)]for (key, value) in data:    result[key].append(value)

关于“python的setdefault与defaultdict函数怎么用”的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识,可以关注亿速云行业资讯频道,小编每天都会为大家更新不同的知识点。

亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

原文链接:https://my.oschina.net/liuzhijun/blog/4621910

AI

开发者交流群×