在 Python 中,我们经常需要处理字典数据。有时,我们需要将多个字典合并成一个字典,并对键值进行累加操作。然而,我们发现,使用普通的字典更新操作(dict.update())时,键值会发生覆盖,而不是累加。
为了解决这个问题,我们可以创建一个自定义的字典类,重写 update() 方法,使之能够对键值进行累加。下面的代码展示了如何实现:
class StatementDict(dict):
def update(self, statement_dict):
for statement in statement_dict.iteritems():
self.add(statement)
def add(self, statement):
ann_id = statement[0]
lvl_dict = statement[1]
if ann_id in self:
self[ann_id]['skill'] += lvl_dict['skill']
self[ann_id]['knowledge'] += lvl_dict['knowledge']
self[ann_id]['interest'] += lvl_dict['interest']
else:
self[ann_id] = lvl_dict
我们使用这个自定义的字典类来对多个字典进行合并和累加操作,并发现结果是正确的,与我们预期的相同。然而,我们也发现,如果我们改变合并字典的顺序,累加结果也会发生变化。这是为什么呢?
2、解决方案
原来,问题出在我们自定义的字典类中。在 add() 方法中,我们直接将 lvl_dict 赋值给 self[ann_id],这导致 self[ann_id] 和 lvl_dict 指向同一个字典对象。当我们对 self[ann_id] 进行修改时,实际上也修改了 lvl_dict。
为了解决这个问题,我们需要在 add() 方法中复制一份 lvl_dict,而不是直接赋值。这样,self[ann_id] 和 lvl_dict 就指向不同的字典对象,对 self[ann_id] 的修改不会影响 lvl_dict。
修改后的代码如下:
class StatementDict(dict):
def update(self, statement_dict):
for statement in statement_dict.iteritems():
self.add(statement)
def add(self, statement):
ann_id = statement[0]
lvl_dict = statement[1]
if ann_id in self:
self[ann_id]['skill'] += lvl_dict['skill']
self[ann_id]['knowledge'] += lvl_dict['knowledge']
self[ann_id]['interest'] += lvl_dict['interest']
else:
self[ann_id] = lvl_dict.copy() # 复制一份 lvl_dict
经过修改后,我们再次对多个字典进行合并和累加操作,发现结果是正确的,并且与合并字典的顺序无关。
此外,我们还发现了另一个问题。当我们使用 update() 方法更新 StatementDict 时,原始的 StatementDict 实例也会发生改变。这是因为 update() 方法会修改 self,而 self 是一个指向 StatementDict 实例的引用。
为了解决这个问题,我们需要在 update() 方法中创建一个新的 StatementDict 实例,并将原有实例中的数据复制到新的实例中。这样,原有实例就不会被修改。
修改后的代码如下:
class StatementDict(dict):
def update(self, statement_dict):
new_dict = StatementDict()
new_dict.update(self)
new_dict.update(statement_dict)
return new_dict
def add(self, statement):
ann_id = statement[0]
lvl_dict = statement[1]
if ann_id in self:
self[ann_id]['skill'] += lvl_dict['skill']
self[ann_id]['knowledge'] += lvl_dict['knowledge']
self[ann_id]['interest'] += lvl_dict['interest']
else:
self[ann_id] = lvl_dict.copy()
经过修改后,我们再次对多个字典进行合并和累加操作,发现结果是正确的,并且与合并字典的顺序无关,也不会修改原始的 StatementDict 实例。