内容简介:在做 code review 的时候,发现有同学的代码看起来不那么 pythonic,忍不住写篇短文介绍两个函数:setdefault 与 defaultdict。有这样的需求,一个列表,里面是(key, value) 这样的键值对元组,要将它转换成一个字典对象,并将key相同的value作为一组。看代码:
在做 code review 的时候,发现有同学的代码看起来不那么 pythonic,忍不住写篇短文介绍两个函数: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]}
下面这个方法是大家都能想到的,先判断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) >>> y 4 >>> d {'x': 3}
对比
>>> y = d.setdefault("y", 4) >>> y 4 >>> 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) >>> result defaultdict(<type 'list'>, {}) >>> result['a'] []
参数为 list,它就会构建一个默认value为list的字典,例如result['a']的值默认就是list对象。
因此,前面这段代码可以改为:
from collections import defaultdict result = 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之禅」(id:vttalk)获取最新文章
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持 码农网
猜你喜欢:- JS变量声明和函数声明提升
- 利用函数组合提升代码可维护性
- 利用函数组合提升代码可维护性
- 20 个 Lodash 函数瞬间提升代码逼格
- 8个超好用的Python内置函数,提升效率必备
- Cadence公布人工智能芯片Tensilica DNA 100,性能提升4.7倍,能耗比提升2.3倍
本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们。