1. 程式人生 > >提升逼格的兩個函式:setdefault 與 defaultdict

提升逼格的兩個函式:setdefault 與 defaultdict

在做 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之禪」(id:vttalk)獲取最新文章 python之禪