1.20 合并多個(gè)字典或映射

2018-02-24 15:26 更新

問題

現(xiàn)在有多個(gè)字典或者映射,你想將它們從邏輯上合并為一個(gè)單一的映射后執(zhí)行某些操作,比如查找值或者檢查某些鍵是否存在。

解決方案

加入你有如下兩個(gè)字典:

a = {'x': 1, 'z': 3 }
b = {'y': 2, 'z': 4 }

現(xiàn)在假設(shè)你必須在兩個(gè)字典中執(zhí)行查找操作(比如先從a中找,如果找不到再在b中找)。一個(gè)非常簡(jiǎn)單扼解決方案就是使用collections模塊中的ChainMap類。比如:

from collections import ChainMap
c = ChainMap(a,b)
print(c['x']) # Outputs 1 (from a)
print(c['y']) # Outputs 2 (from b)
print(c['z']) # Outputs 3 (from a)

討論

一個(gè)ChainMap接受多個(gè)字典并將它們?cè)谶壿嬌献優(yōu)橐粋€(gè)字典。然后,這些字典并不是真的合并在一起了,ChainMap類只是在內(nèi)部創(chuàng)建了一個(gè)容納這些字典的列表并重新定義了一些常見的字典操作來遍歷這個(gè)列表。大部分字典操作都是可以正常使用的,比如:

>>> len(c)
3
>>> list(c.keys())
['x', 'y', 'z']
>>> list(c.values())
[1, 2, 3]
>>>

如果出現(xiàn)重復(fù)鍵,那么第一次出現(xiàn)的映射值會(huì)被返回。因此,例子程序中的c[‘z']總是會(huì)返回字典a中對(duì)應(yīng)的值,而不是b中對(duì)應(yīng)的值。

對(duì)于字典的更新或刪除操作總是影響的是列表中第一個(gè)字典。比如:

>>> c['z'] = 10
>>> c['w'] = 40
>>> del c['x']
>>> a
{'w': 40, 'z': 10}
>>> del c['y']
Traceback (most recent call last):
...
KeyError: "Key not found in the first mapping: 'y'"
>>>

ChainMap對(duì)于編程語(yǔ)言中的作用范圍變量(比如globals, locals等)是非常有用的。事實(shí)上,有一些方法可以使它變得簡(jiǎn)單:

>>> values = ChainMap()
>>> values['x'] = 1
>>> # Add a new mapping
>>> values = values.new_child()
>>> values['x'] = 2
>>> # Add a new mapping
>>> values = values.new_child()
>>> values['x'] = 3
>>> values
ChainMap({'x': 3}, {'x': 2}, {'x': 1})
>>> values['x']
3
>>> # Discard last mapping
>>> values = values.parents
>>> values['x']
2
>>> # Discard last mapping
>>> values = values.parents
>>> values['x']
1
>>> values
ChainMap({'x': 1})
>>>

作為ChainMap的替代,你可能會(huì)考慮使用update()方法將兩個(gè)字典合并。比如:

>>> a = {'x': 1, 'z': 3 }
>>> b = {'y': 2, 'z': 4 }
>>> merged = dict(b)
>>> merged.update(a)
>>> merged['x']
1
>>> merged['y']
2
>>> merged['z']
3
>>>

這樣也能行得通,但是它需要你創(chuàng)建一個(gè)完全不同的字典對(duì)象(或者是破壞現(xiàn)有字典結(jié)構(gòu))。同時(shí),如果原字典做了更新,這種改變不會(huì)反應(yīng)到新的合并字典中去。比如:

>>> a['x'] = 13
>>> merged['x']
1

ChianMap使用原來的字典,它自己不創(chuàng)建新的字典。所以它并不會(huì)產(chǎn)生上面所說的結(jié)果,比如:

>>> a = {'x': 1, 'z': 3 }
>>> b = {'y': 2, 'z': 4 }
>>> merged = ChainMap(a, b)
>>> merged['x']
1
>>> a['x'] = 42
>>> merged['x'] # Notice change to merged dicts
42
>>>
以上內(nèi)容是否對(duì)您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號(hào)
微信公眾號(hào)

編程獅公眾號(hào)