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

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

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

然而，這個方法有一個問題——它會改變其中一個字典。如果我們不想改變原有的兩個字典，那么我們必需要單獨再創(chuàng)建一個字典：

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

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

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

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

使用 ChainMap 可以把多個字典合并成一個 ChainMap 對象。讀寫這個對象就像是讀字典一樣。

例如：

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)行效果如下圖所示：

不僅可以“合并”兩個字典， ChainMap 可以接受任意多個字典，并把他們?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 不會真的把字典合并在一起，而是在內(nèi)部儲存一個Key到每個字典的映射，當(dāng)你讀取 e[key] 的時候，它先去查詢這個key在哪個字典里面，然后再去對應(yīng)的字典里面查詢對應(yīng)的值。所以使用ChainMap幾乎不需要額外的內(nèi)存空間（當(dāng)前這個對象自己會占用一些空間，但是如果要合并大字典，那么它自己占用的空間幾乎可以忽略）。

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

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

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

首先來說第一個問題的答案：

ChainMap 對象會使用第一個擁有這個Key的字典里面的值，如下圖所示：

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

第三個問題，如果修改了原來的字典，那么 ChainMap 對象也會相應(yīng)更新：

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

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

 總結(jié)

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

最新評論