首頁 > 後端開發 > Python教學 > 簡介Python的collections模組中defaultdict類型的用法

簡介Python的collections模組中defaultdict類型的用法

WBOY
發布: 2016-07-22 08:56:29
原創
1868 人瀏覽過

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
登入後複製

來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板