举例讲解Python中字典的合并值相加与异或对比

发表于 5年以前 | 总阅读数：1022 次

字典合并值相加
在统计汇总游戏数据的时候，有些数据是是每天用字典存的，当我要对多天汇总的时候，就需要合并字典了。
如果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]

对字典diff("异或")
在游戏中，我要监控记录物品系统中的背包变动情况。("异或"的结果是相同的消除,剩下不同的,即变动的)
假设背包的存储结构是这样的。
是一个字典，{物品id:数量}。
在背包类初始化的时候，把背包物品信息copy保存到一个oldbag变量，进行一些物品操作后(比如使用物品，领取物品奖励等)，在调用save()方法存进redis时，对新的bag字典与oldbag字典进行差异对比就得出变动情况了。
千言不如一码。


    def symmetric_difference(_oldobj,_newobj): 
      _oldkeys = _oldobj.keys() 
      _newkeys = _newobj.keys() 
      _diff = {} 
      for _key in set(_oldkeys + _newkeys): 
        _val = _newobj.get(_key,0) - _oldobj.get(_key,0) 
        if _val: 
          _diff[_key] = _val  
      return _diff  

    oldobj = {'a':1,'b':2,'c':3} 
    newobj = {'a':1,'b':3,'d':4} 
    print symmetric_difference(oldobj,newobj)

输出