在Python中，當(dāng)我們有兩個(gè)字典需要合并的時(shí)候，可以使用字典的 update 方法，例如：

a = {'a': 1, 'b': 2}
b = {'x': 3, 'y': 4}
a.update(b)
print(a)

運(yùn)行效果如下圖所示：

然而，這個(gè)方法有一個(gè)問(wèn)題——它會(huì)改變其中一個(gè)字典。如果我們不想改變?cè)械膬蓚€(gè)字典，那么我們必需要單獨(dú)再創(chuàng)建一個(gè)字典：

a = {'a': 1, 'b': 2}
b = {'x': 3, 'y': 4}
c = dict(a)
c.update(b)
print(c)
print(a)

如果原來(lái)的兩個(gè)字典非常大，那么這種方式將會(huì)浪費(fèi)大量的內(nèi)存。

無(wú)論是直接修改原有的其中一個(gè)字典，還是創(chuàng)建另一個(gè)字典，這兩種方案都有點(diǎn)缺陷。那么有沒(méi)有既不修改原有字典，又不另外創(chuàng)建一個(gè)新的字典的方法呢？

答案就是 collections 模塊下面的 ChainMap 。

使用 ChainMap 可以把多個(gè)字典合并成一個(gè) ChainMap 對(duì)象。讀寫(xiě)這個(gè)對(duì)象就像是讀字典一樣。

例如：

from collections import ChainMap
a = {'a': 1, 'b': 2}
b = {'x': 3, 'y': 4}
c = ChainMap(a, b)
print(c['a'])
print(c['y'])

運(yùn)行效果如下圖所示：

不僅可以“合并”兩個(gè)字典， ChainMap 可以接受任意多個(gè)字典，并把他們?nèi)己显谝黄穑?/p>

from collections import ChainMap
a = {'a': 1, 'b': 2}
b = {'x': 3, 'y': 4}
c = {'z': 5, 'w': 6}
d = {'m': 7, 'h': 8, 'k': 9}
e = ChainMap(a, b, c, d)
print(e['a'], e['y'], e['z'], e['k'])

運(yùn)行效果如下圖所示：

ChainMap 不會(huì)真的把字典合并在一起，而是在內(nèi)部?jī)?chǔ)存一個(gè)Key到每個(gè)字典的映射，當(dāng)你讀取 e[key] 的時(shí)候，它先去查詢(xún)這個(gè)key在哪個(gè)字典里面，然后再去對(duì)應(yīng)的字典里面查詢(xún)對(duì)應(yīng)的值。所以使用ChainMap幾乎不需要額外的內(nèi)存空間（當(dāng)前這個(gè)對(duì)象自己會(huì)占用一些空間，但是如果要合并大字典，那么它自己占用的空間幾乎可以忽略）。

所以你是不是覺(jué)得使用 ChainMap 就能實(shí)現(xiàn)完美合并字典了呢？

在使用它之前，你一定要理解它的運(yùn)行原理。如果你理解了它的運(yùn)行原理，那么下面幾個(gè)問(wèn)題，你在運(yùn)行代碼之前就會(huì)知道結(jié)果是什么：

如果兩個(gè)字典里面有一個(gè)Key的名字相同，那么使用ChainMap以后會(huì)讀取哪一個(gè)？ 如果為ChainMap對(duì)象添加一個(gè)Key-Value對(duì)，那么這個(gè)值會(huì)添加到哪里？ 如果從原字典里面刪除一個(gè)Key，ChainMap對(duì)象里面的Key也會(huì)消失嗎？ 如果從ChainMap對(duì)象里面刪除一個(gè)Key，那么原字典里面的Key會(huì)消失嗎？

首先來(lái)說(shuō)第一個(gè)問(wèn)題的答案：

ChainMap 對(duì)象會(huì)使用第一個(gè)擁有這個(gè)Key的字典里面的值，如下圖所示：

第二個(gè)問(wèn)題，新的Key-Value會(huì)被添加進(jìn)第一個(gè)字典里面，如下圖所示：

第三個(gè)問(wèn)題，如果修改了原來(lái)的字典，那么 ChainMap 對(duì)象也會(huì)相應(yīng)更新：

第四個(gè)問(wèn)題，如果這個(gè)Key只在一個(gè)源字典中存在，那么這個(gè)Key會(huì)被從源字典中刪除。如果這個(gè)Key在多個(gè)字典中都存在，那么Key會(huì)被從第一個(gè)字典中刪除。當(dāng)被從第一個(gè)字典中刪除以后，第二個(gè)源字典的Key可以繼續(xù)被 ChainMap 讀取。

以上4點(diǎn)，在你使用 ChainMap 的時(shí)候一定要牢記，否則可能會(huì)導(dǎo)致你發(fā)現(xiàn)不了的Bug。 

 總結(jié)

以上所述是小編給大家介紹的在Python中合并字典模塊ChainMap的隱藏坑，希望對(duì)大家有所幫助，如果大家有任何疑問(wèn)歡迎給我留言，小編會(huì)及時(shí)回復(fù)大家的！

最新評(píng)論