洋書、時々プログラミング

博士課程修了→メーカーという経路を辿っている人の日常

python 辞書のキーを毎回設定してくれるdefault dict

pythonで辞書を使っていて、キーの値をいちいち初期化する必要があってダルいと感じていたら、初期化してくれる方法があったのでメモ 方法としては、defaultdictを使う。以下にサンプルコードを置いておく

from collections import defaultdict         # defalt dictのインポート

a = defaultdict(float)                      # dictの初期化
a["test"] = 2                               # testに値を格納

print(a["test"])                            # 初期化後にデータを入れるとその値(ここでは2)が表示される
a = defaultdict(float)                      # 再度初期化
print(a["test"])                            # 初期化することで0.0が代入されている。
a = defaultdict(int)                        # 再度初期化
print(a["test"])                            # 初期化することで0.0が代入されている。

どんな時に使うか

出現語数のカウントなど、単語ごとにキーチェック→カウントという流れをこの方法で簡略化できそう。