【上海校区】Python提升逼格的两个函数：setdefault 与 defaultdict - 黑马程序员技术交流社区

在做 code review 的时候，发现有同学的代码看起来不那么 pythonic，忍不住写篇短文介绍两个函数：setdefault 与 defaultdict。

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

[Python] 纯文本查看 复制代码

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中不怎么优雅

[Python] 纯文本查看 复制代码

result = {}

for (key, value) in data:

    if key in result:

        result[key].append(value)

    else:

        result[key] = [value]

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

[Python] 纯文本查看 复制代码

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设置的默认值会改变原字典的值。

[Python] 纯文本查看 复制代码

>>> d = {"x":3}

>>> y = d.get("y", 4)

>>> y

4

>>> d

{'x': 3}

[Python] 纯文本查看 复制代码

>>> y = d.setdefault("y", 4)

>>> y

4

>>> d

{'y': 4, 'x': 3}

[Python] 纯文本查看 复制代码

result = {}

data = [("p", 1), ("p", 2), ("p", 3),

        ("h", 1), ("h", 2), ("h", 3)]

for (key, value) in data:

    result.setdefault(key, []).append(value)

[Python] 纯文本查看 复制代码

>>> result = defaultdict(list)

>>> result

defaultdict(<type 'list'>, {})

>>> result['a']

[]

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

因此，前面这段代码可以改为：

[Python] 纯文本查看 复制代码

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)