飄逸的python - 字典合並值相加

在統計匯總游戲數據的時候，有些數據是是每天用字典存的，當我要對多天匯總的時候，就需要合並字典了。

如果key相同的話它們的值就相加。

不能用update方法，因為用update方法則相同的key的值會覆蓋，而不是相加。

千言不如一碼。

def union_dict(*objs):
    _keys = set(sum([obj.keys() for obj in objs],[]))
    _total = {}
    for _key in _keys:
        _total[_key] = sum([obj.get(_key,0) for obj in objs])
    return _total

obj1 = {'a':1,'b':2,'c':3}
obj2 = {'a':1,'b':3,'d':4}
print union_dict(obj1,obj2)

輸出

{'a': 2, 'c': 3, 'b': 5, 'd': 4}

 

sum([obj.keys() for obj in objs],[])這句可能不太好理解。

其實sum()函數也有" 鮮為人知的參數",即第2個參數,start參數，默認是0。

而且不止可以是int類型，還可以是其他支持+操作符的東西，比如[]。

利用這一點，可以對二層數組打平成一層。

比如

>>sum([[1,2,3],[4,5]],[])

[1,2,3,4,5]