ある辞書から別の辞書に対応するキーの要素を代入する

Question

タイトルが少し分かりづらいですが、

dict_A = {"a": 0, "b": 0, "c": 0,"d": 0,"e": 0,"f": 0}
dict_B = {"c": 10, "f": 15}

上のような辞書があったとき、

new_dict_A = {"a": 0, "b": 0, "c": 10,"d": 0,"e": 0,"f": 15}

上のようにdict_Aと同じキーに対応する場所にdict_Bの要素を代入し、new_dict_Aを作成したいです。
for文を使えばできるのですが、計算時間がかかってしまうので、出来るだけ計算コストを抑えた方法がありましたら教えていただきたいです。

Answer 1

後から追加した要素が優先されるので、これで充分です。

>>> dict_A = {"a": 0, "b": 0, "c": 0,"d": 0,"e": 0,"f": 0}
>>> dict_B = {"c": 10, "f": 15}
>>>
>>> new_dict_A = {**dict_A, **dict_B}
>>> new_dict_A
{'a': 0, 'b': 0, 'c': 10, 'd': 0, 'e': 0, 'f': 15}

ただし破壊的な方法でも良いならdict.updateの方が早そうです。

Comments

[hayataka2049]

この辺の仕様って明文化されているのでしょうか……ドキュメントにすら書いてない実装依存とかだと嫌だなぁ、と

[LouiS0616]

> 6.2.7. 辞書表示

> キーとデータからなる対の並びがカンマ区切りで与えられたときは、その要素は左から右へ評価され、辞書のエントリーを定義します。すなわち、それぞれのキーオブジェクトが、辞書内で対応するデータを保存するキーとして使われます。 これにより、キーとデータのリストの中で同じキーを複数回指定することができ、そのキーに対する最終的な辞書の値は、最後に与えられたものになります。

> ダブルアスタリスク ** は 辞書のアンパック を表します。このとき被演算子は mapping でなければなりません。それぞれのmappingの要素は、新たな辞書に追加されます。キー/データの対もしくは辞書のアンパックによって先に追加された値は、後から追加された値によって上書きされます。

https://docs.python.org/ja/3/reference/expressions.html#dict

---
> In dictionaries, later values will always override earlier ones

https://www.python.org/dev/peps/pep-0448/

[hayataka2049]

ありがとうございます。納得しました。

Answer 2

dict.updateが使えますが、単純に使うとin-placeで処理されて元の辞書が変更されるので、必要に応じてcopyすると良いと思います。

dict_A = {"a": 0, "b": 0, "c": 0,"d": 0,"e": 0,"f": 0}
dict_B = {"c": 10, "f": 15}

new_dict_A = dict_A.copy()
new_dict_A.update(dict_B)
print(dict_A)
print(new_dict_A)
""" =>
{'a': 0, 'b': 0, 'c': 0, 'd': 0, 'e': 0, 'f': 0}
{'a': 0, 'b': 0, 'c': 10, 'd': 0, 'e': 0, 'f': 15}
"""

で、要素数にもよりますが、ループで書いてもそんなに遅くはないかと。

import timeit

dict_A = {"a": 0, "b": 0, "c": 0,"d": 0,"e": 0,"f": 0}
dict_B = {"c": 10, "f": 15}

def f1():
    new_dict_A = dict_A.copy()
    for k, v in dict_B.items():
        new_dict_A[k] = v
    return new_dict_A

def f2():
    new_dict_A = dict_A.copy()
    new_dict_A.update(dict_B)
    return new_dict_A

print(timeit.timeit(f1))  # => 0.5886494450096507
print(timeit.timeit(f2))  # => 0.46052658899861854

元々ない要素は更新しないとか、柔軟なことをやりたければループの方が向いているかもしれません。

Answer 3

(きっちりと実装を読んでないのですが) updateメソッドを使えば効率的に実行できます。

dict_A.update(dict_B)

print(dict_A)  # {'a': 0, 'b': 0, 'c': 10, 'd': 0, 'e': 0, 'f': 15}

jupyter notebookで評価した結果をつけておきます。
https://gist.github.com/yuokada/e7cc0a9b0552079e5bdb1ae9c2446ab4

Answer 4

updateで更新できます。これで良かったでしょうか？

dict_A = {"a":0,"b":0,"c":0,"d":0,"e":0,"f":0}
dict_B = {"c":10,"f":15}

dict_A.update(dict_B)
print(dict_A)