defaultdict 主要用來需要對 value 做初始化的情形。對於字典來說,key 必須是 hashable,immutable,unique 的數據,而 value 可以是任意的數據類型。如果 value 是 list,dict 等數據類型,在使用之前必須初始化為空,有些情況需要把 value 初始化為特殊值,比如 0 或者 ‘‘。
from collections import defaultdict person_by_age = defaultdict(list) for person in persons: d[person.age].append(person.name)
defaultdict 和 dict 的使用方法一樣,只有在初始化的時候必須傳入一個 callable 的對象 x,當訪問某個還不存在的 key 時,會把 value 自動設置成 x()。比如上例中,當第一次訪問某個年齡的人 d[person.age] 就會變成 list(), 也就是 []。
當然也可以使用自己定義的 callable 對象,比如:
d = defaultdict(lambda: 0) d["hello"] += 1 # 1 d["a"] # 0
defaultdict 要比 dict.set_default 效率更高,使用起來也更直觀和方便。
標準字典包括一個方法setdefault()來獲取一個值,如果值不存在則建立一個默認值。defaultdict初始化容器是會讓調用者提前指定默認值。
import collections def default_factory(): return 'default value' d = collections.defaultdict(default_factory, foo = 'bar') print 'd:', d print 'foo =>', d['foo'] print 'var =>', d['bar']
只要所有鍵都有相同的默認值,就可以使用這個方法。
上面的結果是:
d: defaultdict(<function default_factory at 0x0201FAB0>, {'foo': 'bar'}) foo => bar var => default value
Tags: function default person return import
文章來源: