デフォルト辞書
「標準辞書には、値を取得し、値が存在しない場合にデフォルトを設定するための setdefault() メソッドが含まれています。対照的defaultdict
に、コンテナが初期化されるときに、呼び出し元がデフォルト (返される値) を事前に指定できるようにします。」
The Python Standard Library by ExampleでDoug Hellmannによって定義されている
デフォルトディクトの使い方
defaultdict のインポート
>>> from collections import defaultdict
defaultdict を初期化する
渡して初期化する
最初の引数としてcallable (必須)
>>> d_int = defaultdict(int)
>>> d_list = defaultdict(list)
>>> def foo():
... return 'default value'
...
>>> d_foo = defaultdict(foo)
>>> d_int
defaultdict(<type 'int'>, {})
>>> d_list
defaultdict(<type 'list'>, {})
>>> d_foo
defaultdict(<function foo at 0x7f34a0a69578>, {})
** 2 番目の引数としてのkwargs (オプション)
>>> d_int = defaultdict(int, a=10, b=12, c=13)
>>> d_int
defaultdict(<type 'int'>, {'a': 10, 'c': 13, 'b': 12})
また
>>> kwargs = {'a':10,'b':12,'c':13}
>>> d_int = defaultdict(int, **kwargs)
>>> d_int
defaultdict(<type 'int'>, {'a': 10, 'c': 13, 'b': 12})
それはどのように機能しますか
標準辞書の子クラスと同様に、すべて同じ機能を実行できます。
ただし、不明なキーを渡すと、エラーではなくデフォルト値が返されます。例:
>>> d_int['a']
10
>>> d_int['d']
0
>>> d_int
defaultdict(<type 'int'>, {'a': 10, 'c': 13, 'b': 12, 'd': 0})
デフォルト値を変更したい場合は、default_factory を上書きします:
>>> d_int.default_factory = lambda: 1
>>> d_int['e']
1
>>> d_int
defaultdict(<function <lambda> at 0x7f34a0a91578>, {'a': 10, 'c': 13, 'b': 12, 'e': 1, 'd': 0})
また
>>> def foo():
... return 2
>>> d_int.default_factory = foo
>>> d_int['f']
2
>>> d_int
defaultdict(<function foo at 0x7f34a0a0a140>, {'a': 10, 'c': 13, 'b': 12, 'e': 1, 'd': 0, 'f': 2})
質問の例
例 1
int が default_factory として渡されているため、不明なキーはデフォルトで 0 を返します。
文字列がループに渡されると、d のアルファベットの数が増えます。
>>> s = 'mississippi'
>>> d = defaultdict(int)
>>> d.default_factory
<type 'int'>
>>> for k in s:
... d[k] += 1
>>> d.items()
[('i', 4), ('p', 2), ('s', 4), ('m', 1)]
>>> d
defaultdict(<type 'int'>, {'i': 4, 'p': 2, 's': 4, 'm': 1})
例 2
リストが default_factory として渡されているため、不明な (存在しない) キーはデフォルトで [ ] (つまりリスト) を返します。
タプルのリストがループで渡されると、d[color] に値が追加されます
>>> s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
>>> d = defaultdict(list)
>>> d.default_factory
<type 'list'>
>>> for k, v in s:
... d[k].append(v)
>>> d.items()
[('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]
>>> d
defaultdict(<type 'list'>, {'blue': [2, 4], 'red': [1], 'yellow': [1, 3]})