在Python中合併字典模塊ChainMap的隱藏坑【推薦】

在Python中,當我們有兩個字典需要合併的時候,可以使用字典的 update 方法,接下來通過本文給大家介紹在Python中合併字典模塊ChainMap的隱藏坑,感興趣的朋友一起看看吧

在Python中,當我們有兩個字典需要合併的時候,可以使用字典的 update 方法,例如:

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

運行效果如下圖所示:

然而,這個方法有一個問題——它會改變其中一個字典。如果我們不想改變原有的兩個字典,那麼我們必需要單獨再創建一個字典:

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

如果原來的兩個字典非常大,那麼這種方式將會浪費大量的內存。

無論是直接修改原有的其中一個字典,還是創建另一個字典,這兩種方案都有點缺陷。那麼有沒有既不修改原有字典,又不另外創建一個新的字典的方法呢?

答案就是 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'])

運行效果如下圖所示:

不僅可以“合併”兩個字典, ChainMap 可以接受任意多個字典,並把他們全都合在一起:

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'])

運行效果如下圖所示:

ChainMap 不會真的把字典合併在一起,而是在內部儲存一個Key到每個字典的映射,當你讀取 e[key] 的時候,它先去查詢這個key在哪個字典裏面,然後再去對應的字典裏面查詢對應的值。所以使用ChainMap幾乎不需要額外的內存空間(當前這個對象自己會佔用一些空間,但是如果要合併大字典,那麼它自己佔用的空間幾乎可以忽略)。

所以你是不是覺得使用 ChainMap 就能實現完美合併字典了呢?

在使用它之前,你一定要理解它的運行原理。如果你理解了它的運行原理,那麼下面幾個問題,你在運行代碼之前就會知道結果是什麼:

如果兩個字典裏面有一個Key的名字相同,那麼使用ChainMap以後會讀取哪一個? 如果爲ChainMap對象添加一個Key-Value對,那麼這個值會添加到哪裏? 如果從原字典裏面刪除一個Key,ChainMap對象裏面的Key也會消失嗎? 如果從ChainMap對象裏面刪除一個Key,那麼原字典裏面的Key會消失嗎?

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

ChainMap 對象會使用第一個擁有這個Key的字典裏面的值,如下圖所示:

第二個問題,新的Key-Value會被添加進第一個字典裏面,如下圖所示:

第三個問題,如果修改了原來的字典,那麼 ChainMap 對象也會相應更新:

第四個問題,如果這個Key只在一個源字典中存在,那麼這個Key會被從源字典中刪除。如果這個Key在多個字典中都存在,那麼Key會被從第一個字典中刪除。當被從第一個字典中刪除以後,第二個源字典的Key可以繼續被 ChainMap 讀取。

以上4點,在你使用 ChainMap 的時候一定要牢記,否則可能會導致你發現不了的Bug。 

 總結

以上所述是小編給大家介紹的在Python中合併字典模塊ChainMap的隱藏坑,希望對大家有所幫助,如果大家有任何疑問歡迎給我留言,小編會及時回覆大家的!

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章