本文和大家分享的是python中使用一行代碼優(yōu)雅合并兩個(gè)dict相關(guān)內(nèi)容,一起來(lái)看看吧,希望對(duì)大家學(xué)習(xí)python有所幫助。
一行代碼合并兩個(gè)dict
假設(shè)有兩個(gè)dict x和y,合并成一個(gè)新的dict,不改變 x和y的值,例如
x = {'a': 1, 'b': 2}
y = {'b': 3, 'c': 4}
期望得到一個(gè)新的結(jié)果Z,如果key相同,則y覆蓋x。期望的結(jié)果是
>>> z
{'a': 1, 'b': 3, 'c': 4}
在PEP448中,有個(gè)新的語(yǔ)法可以實(shí)現(xiàn),并且在python3.5中支持了該語(yǔ)法,合并代碼如下
z = {**x, **y}
妥妥的一行代碼。 由于現(xiàn)在很多人還在用python2,對(duì)于python2和python3.0-python3.4的人來(lái)說(shuō),有一個(gè)比較優(yōu)雅的方法,但是需要兩行代碼。
z = x.copy()
z.update(y)
上面的方法,y都會(huì)覆蓋x里的內(nèi)容,所以最終結(jié)果b=3.
不使用python3.5如何一行完成了
如果您還沒(méi)有使用Python 3.5,或者需要編寫向后兼容的代碼,并且您希望在單個(gè)表達(dá)式中運(yùn)行,則最有效的方法是將其放在一個(gè)函數(shù)中:
def merge_two_dicts(x, y):
"""Given two dicts, merge them into a new dict as a shallow copy."""
z = x.copy()
z.update(y)
return z
然后一行代碼完成調(diào)用:
z = merge_two_dicts(x, y)
你也可以定義一個(gè)函數(shù),合并多個(gè)dict,例如
def merge_dicts(*dict_args):
"""
Given any number of dicts, shallow copy and merge into a new dict,
precedence goes to key value pairs in latter dicts.
"""
result = {}
for dictionary in dict_args:
result.update(dictionary)
return result
然后可以這樣使用
z = merge_dicts(a, b, c, d, e, f, g)
所有這些里面,相同的key,都是后面的覆蓋前面的。
一些不夠優(yōu)雅的示范
items
有些人會(huì)使用這種方法:
z = dict(x.items() + y.items())
這其實(shí)就是在內(nèi)存中創(chuàng)建兩個(gè)列表,再創(chuàng)建第三個(gè)列表,拷貝完成后,創(chuàng)建新的dict,刪除掉前三個(gè)列表。這個(gè)方法耗費(fèi)性能,而且對(duì)于python3,這個(gè)無(wú)法成功執(zhí)行,因?yàn)?/span>items()返回是個(gè)對(duì)象。
>>> c = dict(a.items() + b.items())
Traceback (most recent call last):
File "", line 1, in
TypeError: unsupported operand type(s) for +: 'dict_items' and 'dict_items'
你必須明確的把它強(qiáng)制轉(zhuǎn)換成list,z = dict(list(x.items()) + list(y.items())),這太浪費(fèi)性能了。 另外,想以來(lái)于items()返回的list做并集的方法對(duì)于python3來(lái)說(shuō)也會(huì)失敗,而且,并集的方法,導(dǎo)致了重復(fù)的key在取值時(shí)的不確定,所以,如果你對(duì)兩個(gè)dict合并有優(yōu)先級(jí)的要求,這個(gè)方法就徹底不合適了。
>>> x = {'a': []}
>>> y = {'b': []}
>>> dict(x.items() | y.items())
Traceback (most recent call last):
File "", line 1, in
TypeError: unhashable type: 'list'
這里有一個(gè)例子,其中y應(yīng)該具有優(yōu)先權(quán),但是由于任意的集合順序,x的值被保留:
>>> x = {'a': 2}>>> y = {'a': 1}>>> dict(x.items() | y.items())
{'a': 2}
構(gòu)造函數(shù)
也有人會(huì)這么用
z = dict(x, **y)
這樣用很好,比前面的兩步的方法高效多了,但是可閱讀性差,不夠pythonic,如果當(dāng)key不是字符串的時(shí)候,python3中還是運(yùn)行失敗
>>> c = dict(a, **b)
Traceback (most recent call last):
File "", line 1, in <module>TypeError: keyword arguments must be strings
Guido van Rossum 大神說(shuō)了:宣告dict({}, {1:3})是非法的,因?yàn)楫吘故菫E用 機(jī)制。雖然這個(gè)方法比較hacker,但是太投機(jī)取巧了。
一些性能較差但是比較優(yōu)雅的方法
下面這些方法,雖然性能差,但也比items方法好多了。并且支持優(yōu)先級(jí)。
{k: v for d in dicts for k, v in d.items()}
python2.6中可以這樣
dict((k, v) for d in dicts for k, v in d.items())
itertools.chain 將以正確的順序?qū)㈡I值對(duì)上的迭代器鏈接:
import itertools
z = dict(itertools.chain(x.iteritems(), y.iteritems()))
性能測(cè)試
以下是在Ubuntu 14.04上完成的,在Python 2.7(系統(tǒng)Python)中:
>>> min(timeit.repeat(lambda: merge_two_dicts(x, y)))0.5726828575134277
>>> min(timeit.repeat(lambda: {k: v for d in (x, y) for k, v in d.items()} ))1.163769006729126
>>> min(timeit.repeat(lambda: dict(itertools.chain(x.iteritems(),y.iteritems()))))1.1614501476287842
>>> min(timeit.repeat(lambda: dict((k, v) for d in (x, y) for k, v in d.items())))2.2345519065856934
在python3.5中
>>> min(timeit.repeat(lambda: {**x, **y}))0.4094954460160807
>>> min(timeit.repeat(lambda: merge_two_dicts(x, y)))0.7881555100320838
>>> min(timeit.repeat(lambda: {k: v for d in (x, y) for k, v in d.items()} ))1.4525277839857154
>>> min(timeit.repeat(lambda: dict(itertools.chain(x.items(), y.items()))))2.3143140770262107
>>> min(timeit.repeat(lambda: dict((k, v) for d in (x, y) for k, v in d.items())))3.2069112799945287
來(lái)源:異步社區(qū)