字典合併值相加
在統計匯總遊戲資料的時候,有些資料是是每天用字典存的,當我要對多天總結的時候,就需要合併字典了。
如果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()函數也有"鮮為人知的參數",即第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)
#
{'b': 1, 'd': 4,'c': -3}