Python實用技法第5篇:一鍵多值字典:defaultdict

上一篇文章: Python實用技法第4篇:實現優先級隊列
上一篇文章: Python實用技法第6篇:讓字典保持有序

一、需求🙀

咱們想要一個能將鍵(key)映射到多個值的字(即所謂的一鍵多值字典)

二、解決方案😸

字典是一種關聯容器,每一個鍵都映射到一個單獨的值上。若是想讓鍵映射到多個值,須要將這些多個值保存到另外一個容器(列表或者集合)中。segmentfault

能夠這樣建立字典:app

d={
‘a’:[1,2,3],
'b':[4,5]
}

或者這樣建立:code

d={
'a':{1,2,3},
'b':{4,5}
}

要使用列表仍是集合徹底取決應用的意圖。若是但願保留元素插入的順序,就用列表,若是但願消除重複元素(而且不在乎他們的排序),就用集合。排序

爲了能方便的建立這樣的字典,能夠利用collections模塊中的defaultdict類。defaultdict的一個特色就是它會自動初始化第一個值,這樣只需關注添加元素便可:隊列

from collections import defaultdict

d=defaultdict(list)
d['a'].append(1)
d['a'].append(2)
d['b'].append(4)

print(d)


d=defaultdict(set)
d['a'].add(1)
d['a'].add(2)
d['b'].add(4)

print(d)

結果:get

defaultdict(<class 'list'>, {'a': [1, 2], 'b': [4]})
defaultdict(<class 'set'>, {'a': {1, 2}, 'b': {4}})
上一篇文章: Python實用技法第4篇:實現優先級隊列
上一篇文章: Python實用技法第6篇:讓字典保持有序
相關文章
相關標籤/搜索