Pythonで入れ子の辞書の更新をする関数の書き方

Question

Python で入れ子状態の辞書の、N 階層目の値を更新する関数を書きたいです:

入れ子状態の辞書 dict、および、そのキーのすべて key1, key2, ..., keyN、および、更新する値 val、を与えることで、該当する dict の要素 dict[key1][key2]...[keyN] を更新する以下のような関数を定義したいのですが、どうすればいいでしょうか?

def update_dict(dict, *keys, val):
    dict[key1][key2]...[keyN] = val  # ここの左辺の書き方が分からない。

一般的な入れ子の dict を仮定して、キーの数 N は可変であるとします。更新後の値は val=x という形で明示的に渡します。key が可変のため、どのように内部の手続きを書けば良いか分からず困っています。

Answer 1

再帰を使うと比較的容易に書けそうです。辞書は適切に初期値が与えられていることが前提です。

from pprint import pprint

dct = {"a": {"aa": {"aaa":1,"aab":2},
             "ab": {"aba":3, "abb":4, "abc":5}},
       "b": {"ba":6, "bb":7}}

def update_dict(d, keys, value):
    if len(keys) == 1:
        d[keys[0]] = value
    else:
        update_dict(d[keys[0]], keys[1:], value)

pprint(dct)
update_dict(dct, ("a", "aa", "aaa"), 2018)
pprint(dct)
""" =>
{'a': {'aa': {'aaa': 1, 'aab': 2}, 'ab': {'aba': 3, 'abb': 4, 'abc': 5}},
 'b': {'ba': 6, 'bb': 7}}
{'a': {'aa': {'aaa': 2018, 'aab': 2}, 'ab': {'aba': 3, 'abb': 4, 'abc': 5}},
 'b': {'ba': 6, 'bb': 7}}
"""

Comments

[masayan24]

ご回答ありがとうございました。解決しました!

Answer 2

ループで書いてみました。
ベースはhayataka2049さんのコードを利用しています。

from pprint import pprint

dct = {"a": {"aa": {"aaa":1,"aab":2},
             "ab": {"aba":3, "abb":4, "abc":5}},
       "b": {"ba":6, "bb":7},
       "c":8}

def update_dict(d, keys, value):
    keys = list(keys)
    key = keys.pop(0) # 先頭を削除して取得
    while len(keys) > 0:
        d = d[key]
        key = keys.pop(0)
    d[key] = value

pprint(dct)
update_dict(dct, ("a","aa","aaa"), 111)
update_dict(dct, ("b","bb"), 222)
update_dict(dct, ("c"), 333)
pprint(dct)
"""
{'a': {'aa': {'aaa': 111, 'aab': 2}, 'ab': {'aba': 3, 'abb': 4, 'abc': 5}},
 'b': {'ba': 6, 'bb': 222},
 'c': 333}
"""

Comments

[masayan24]

ご回答ありがとうございました。解決しました。
ループの分効率が良いので、ベストアンサーに選ばさせていただきます。

[masayan24]

最後の代入は、
d[key] = value
の形で行わなければ辞書に反映されない点がポイントだと自分でやって分かりました。
つまり、最後のところで、
d = d[key]
d = value
とするとうまくいかない。
mutable -> immutable になるからでしょうね。
大変参考になりました。