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

2023-02-24 06:01:39

Python字典設定預設值

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

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

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

初始化的時候設定預設值（defaultdict 或 dict.fromkeys）

>>> from collections import defaultdict
>>> aa = defaultdict(int)
>>> aa['a'] = 1
>>> aa['b'] = 2
>>> aa
defaultdict(<class 'int'>, {'a': 1, 'b': 2})
>>> aa['c']
0
>>> aa
defaultdict(<class 'int'>, {'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

獲取值之前的時候設定預設值（setdefault(key, default)）

這裡有個比較特殊的點：只要對應的 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'

獲取值的時候設定預設值（dict.get(key, default)）

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

python建立帶預設值的字典

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

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

總結

以上為個人經驗，希望能給大家一個參考，也希望大家多多支援it145.com。