python技巧(setdefault 与 defaultdict)

转自:https://mp.weixin.qq.com/s/DFF8sRSW7W4nqB6mmLr2jQ

从题目引出

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

一般都会用的方法:

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

setdefault
setdefault是字典对象的一个实例方法,接收两个参数,用法和字典的get方法类似,但是比 get 更强大。 它可以为给字典的key设定一个默认值(如果key不在字典中的时候)
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)