字典合并值相加
在统计汇总游戏数据的时候,有些数据是是每天用字典存的,当我要对多天汇总的时候,就需要合并字典了。
如果key相同的话它们的值就相加。
不能用update方法,因为用update方法则相同的key的值会覆盖,而不是相加。
千言不如一码。
1
2
3
4
5
6
7
8
9
10
|
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) |
输出
1
|
{'a': 2, 'c': 3, 'b': 5, 'd': 4} |
sum([obj.keys() for obj in objs],[])这句可能不太好理解。
其实sum()函数也有"鲜为人知的参数",即第2个参数,start参数,默认是0。
而且不止可以是int类型,还可以是其他支持+操作符的东西,比如[]。
利用这一点,可以对二层数组打平成一层。
比如
1
2
|
>> sum ([[ 1 , 2 , 3 ],[ 4 , 5 ]],[]) [ 1 , 2 , 3 , 4 , 5 ] |
对字典diff("异或")
在游戏中,我要监控记录物品系统中的背包变动情况。("异或"的结果是相同的消除,剩下不同的,即变动的)
假设背包的存储结构是这样的。
是一个字典,{物品id:数量}。
在背包类初始化的时候,把背包物品信息copy保存到一个oldbag变量,进行一些物品操作后(比如使用物品,领取物品奖励等),在调用save()方法存进redis时,对新的bag字典与oldbag字典进行差异对比就得出变动情况了。
千言不如一码。
1
2
3
4
5
6
7
8
9
10
11
12
13
|
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) |
输出
1
|
{'b': 1, 'd': 4,'c': -3} |
代表玩家得到了1个'b'物品,4个'd'物品,失去了3个'c'物品。