在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的隱藏坑,希望對大家有所幫助,如果大家有任何疑問歡迎給我留言,小編會及時回覆大家的!