Python優(yōu)雅的合并兩個(gè)Dict

darkbaby123 發(fā)布于2019-07-30 14:26 / 3438人閱讀

摘要：一些不夠優(yōu)雅的示范有些人會(huì)使用這種方法這其實(shí)就是在內(nèi)存中創(chuàng)建兩個(gè)列表，再創(chuàng)建第三個(gè)列表，拷貝完成后，創(chuàng)建新的，刪除掉前三個(gè)列表。并且支持優(yōu)先級(jí)。

一行代碼合并兩個(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，或者需要編寫(xiě)向后兼容的代碼，并且您希望在單個(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)閕tems()返回是個(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 
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:

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)我的博客逛逛了