Python中如何給字典設定預設值

我們都知道，在 Python 的字典裡邊，如果 key 不存在的話，透過 key 去取值是會報錯的。

>>> aa = {'a':1, 'b':2}
>>> aa['c']
Traceback (most recent call last):
  File "", line 1, inKeyError: 'c'

如果我們在取不到值的時候不報錯而是給定一個預設值的話就友好多了。

>>> from collections import defaultdict
>>> aa = defaultdict(int)
>>> aa['a'] = 1
>>> aa['b'] = 2
>>> aa
defaultdict(, {'a': 1, 'b': 2})
>>> aa['c']
>>> aa
defaultdict(, {'a': 1, 'b': 2, 'c': 0})
>>> aa = dict.fromkeys('abc', 0)
>>> aa
{'a': 0, 'b': 0, 'c': 0}

defaultdict(default_factory) 中的 default_factory 也可以傳入自定義的匿名函式之類的喲。

>>> aa = defaultdict(lambda : 1)
>>> aa['a']
1

這裡有個比較特殊的點：只要對應的 key 已經被設定了值之後，那麼對相同 key 再次設定預設值就沒用了。

因此，如果你在迴圈裡邊給一個 key 重複設定預設值的話，那麼也只會第一次設定的生效。

>>> aa = {'a':1, 'b':2}
>>> aa
{'a': 1, 'b': 2}
>>> aa.get('c')
>>> aa.setdefault('c', 'hello')
'hello'
>>> aa.get('c')
'hello'
>>> aa
{'a': 1, 'b': 2, 'c': 'hello'}
>>> aa.setdefault('c', 'world')
'hello'
>>> aa.get('c')
'hello'

>>> aa = {'a':1, 'b':2}
>>> aa
{'a': 1, 'b': 2}
>>> aa['c']
Traceback (most recent call last):
  File "", line 1, inKeyError: 'c'
>>> aa.get('c')
>>> aa
{'a': 1, 'b': 2}
>>> aa.get('c', 'hello')
'hello'
>>> aa.get('b')
2

防止keyerror建立帶預設值的字典

from collections import defaultdict
data = collections.defaultdict(lambda :[])