Python優雅的合併兩個Dict

一行代碼合併兩個dict

假設有兩個dict x和y，合併成一個新的dict，不改變 x和y的值，例如

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

指望獲得一個新的結果Z，若是key相同，則y覆蓋x。指望的結果是

>>> z
{'a': 1, 'b': 3, 'c': 4}

在PEP448中，有個新的語法能夠實現，而且在python3.5中支持了該語法，合併代碼以下

z = {**x, **y}

妥妥的一行代碼。
因爲如今不少人還在用python2，對於python2和python3.0-python3.4的人來講，有一個比較優雅的方法，可是須要兩行代碼。

z = x.copy()
z.update(y)

上面的方法，y都會覆蓋x裏的內容，因此最終結果b=3.

不使用python3.5如何一行完成了

若是您尚未使用Python 3.5，或者須要編寫向後兼容的代碼，而且您但願在單個表達式中運行，則最有效的方法是將其放在一個函數中：

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

而後一行代碼完成調用:

z = merge_two_dicts(x, y)

你也能夠定義一個函數，合併多個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，都是後面的覆蓋前面的。

一些不夠優雅的示範

items

有些人會使用這種方法：

z = dict(x.items() + y.items())

這其實就是在內存中建立兩個列表，再建立第三個列表，拷貝完成後，建立新的dict，刪除掉前三個列表。這個方法耗費性能，並且對於python3，這個沒法成功執行，由於items()返回是個對象。

>>> c = dict(a.items() + b.items())
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: unsupported operand type(s) for +: 'dict_items' and 
'dict_items'

你必須明確的把它強制轉換成list，z = dict(list(x.items()) + list(y.items()))，這太浪費性能了。
另外，想以來於items()返回的list作並集的方法對於python3來講也會失敗，並且，並集的方法，致使了重複的key在取值時的不肯定，因此，若是你對兩個dict合併有優先級的要求，這個方法就完全不合適了。

>>> x = {'a': []}
>>> y = {'b': []}
>>> dict(x.items() | y.items())
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: unhashable type: 'list'

這裏有一個例子，其中y應該具備優先權，可是因爲任意的集合順序，x的值被保留：

>>> x = {'a': 2}
>>> y = {'a': 1}
>>> dict(x.items() | y.items())
{'a': 2}

構造函數

也有人會這麼用

z = dict(x, **y)

這樣用很好，比前面的兩步的方法高效多了，可是可閱讀性差，不夠pythonic，若是當key不是字符串的時候，python3中仍是運行失敗

>>> c = dict(a, **b)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: keyword arguments must be strings

Guido van Rossum 大神說了：宣告dict（{}， {1：3}）是非法的，由於畢竟是濫用機制。雖然這個方法比較hacker，可是太投機取巧了。

一些性能較差可是比較優雅的方法

下面這些方法，雖然性能差，但也比items方法好多了。而且支持優先級。

{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()))

性能測試

如下是在Ubuntu 14.04上完成的，在Python 2.7（系統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

爲何不來個人博客逛逛了