提升逼格的两个函数:setdefault 与 defaultdict

By 刘志军 , 2018-10-15, 分类: python技术

python

在做 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]}

download.png

下面这个方法是大家都能想到的,先判断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之禅」,回复「1024」免费获取Python资源

python之禅

猜你喜欢

2020-06-07
python合并两个字典
2019-05-12
Python时间戳转日期时间
2017-08-01
Python 2 和 Python 3 主要区别有哪些(一)?
2017-11-08
Python每日一题:第4题:用Python实现斐波那契数列
2017-11-03
Python陷阱:为什么不能用可变对象作为默认参数的值
2017-10-22
图解Python变量与赋值
2020-06-04
object is not subscriptable 是什么错误
2020-06-07
求求你,别再手工造假数据了,fake了解一下
2017-10-29
Python每日一题:第1题
2019-05-07
Python如何提取url中的参数