python dictの使い方

概要

• pythno3からdictの順序が保存されるようになった
  • 二分木での実装に変更された？
• 順序があるdictはpopitem関数があり、LIFOである
• setdefault関数があり、キーがない場合に入れる値である
  • 応用すると、トライ木が簡潔に実装できる
• getは第2引数に、キーが存在しない場合、返す値を設定できる
  • 環境変数が存在しない場合にデフォルト値を返したい時に便利
    • e.g. os.environ.get("DISABLE_TQDM", False)
• dictにデフォルト値をもたせたいときは、collections.defaultdictを用いる
  • collections.defaultdictは名前空間の関係でpickleでシリアライズできない
  • collections.defaultdictはdictの初期化引数に入れることで、Dict[Any, Any]に変換できる

具体例

d = dict(k2=1, k1=2)

assert(d == dict(k1=2, k2=1))
assert(d.popitem() == ("k1", 2))

d["k1"] = 2
assert(d.setdefault("k3", dict()) == {})
assert(d.pop("k1") == 2)

トライ木の実装

WORD_KEY = '$'

trie = {}
for word in words:
    node = trie
    for char in word:
        node = node.setdefault(char, {})
    # mark the existence of a word in trie node
    node[WORD_KEY] = word

typingの応用

• dict, defaultdictともに専用のタイピングが用意されている
  • それぞれに応じた型ヒントを与えれば良い

from typing import Dict, DefaultDict
import collections

nd: Dict = dict()
dd: DefaultDict = collections.defaultdict(int)

参考

• 212. Word Search II/LeetCode