問(wèn)題
現(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類(lèi)。比如:
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類(lèi)只是在內(nèi)部創(chuàng)建了一個(gè)容納這些字典的列表并重新定義了一些常見(jiàn)的字典操作來(lái)遍歷這個(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使用原來(lái)的字典,它自己不創(chuàng)建新的字典。所以它并不會(huì)產(chǎn)生上面所說(shuō)的結(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 >>>
推薦教程:《Python教程》