飘逸的python - 重载操作符

前面有篇博文《 飘逸的python - 字典合并值相加》讲到了如何对字典合并。

通过调用union_dict(obj1,obj2) 来合并。

在这里，我们重载操作符直接用obj1 + obj2来到达相同的目的。

实现方法就是，定义一个类，直接继承dict，没错，直接继承内置数据结构。

然后定义魔术方法__add__()。

下面看代码。

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

class ddict(dict):
    def __add__(self,obj):
        return union_dict(self,obj)

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

obj1 = ddict(obj1 )
print obj1 + obj2 

输出

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