从字典中减去元素

eSe*_*ano 1 python counter dictionary

我有两本词典,让我们说:

dict_a = {"A": 3.0, "C": 4.0, "B": 6.0}
dict_b = {"A": 3.0, "C": 5.0, "D": 1.0}
Run Code Online (Sandbox Code Playgroud)

我想计算一个减法,dict_a - dict_b所以输出给了我:

{"A": 0.0, "B": 6.0, "C": -1.0, "D": -1.0}
Run Code Online (Sandbox Code Playgroud)

谷歌搜索了一段时间我见过人们使用Counter来执行这样的操作,但做了一些简单的事情

dict(Counter(dict_a) - Counter(dict_b))
Run Code Online (Sandbox Code Playgroud)

只给了我

{"A": 0.0, "B": 6.0}
Run Code Online (Sandbox Code Playgroud)

即,如果键不在dict_a,它将不会出现在输出中.

到目前为止,我已经管理了以下解决方案:

dict_b2 = dict_b.copy()
dict_c = {}
for i in dict_a.keys():
   dict_c.update({i: dict_a[i] - dict_b2.pop(i,0.0)})
   # Changes sign to the remaining values from dict_b2
dict_c.update({k: 0.0 - v for k, v in dict_b2.iteritems()})
Run Code Online (Sandbox Code Playgroud)

但我知道必须有一种更加优雅和有效的方式.

有任何想法吗?

mgi*_*son 7

事实证明,Counter有一个(没有很好的文件记录)subtract方法:

>>> dict_a = {"A": 3.0, "C": 4.0, "B": 6.0}
>>> dict_b = {"A": 3.0, "C": 5.0, "D": 1.0}
>>> from collections import Counter
>>> a = Counter(dict_a)
>>> a.subtract(dict_b)
>>> a
Counter({'B': 6.0, 'A': 0.0, 'C': -1.0, 'D': -1.0})
Run Code Online (Sandbox Code Playgroud)

请注意,这不会受到强加的"无0值以下"约束的影响Counter.__sub__,但它确实可以就地进行操作.

有点酷的是,我没有必要将两个 dicts 转换为Counter - 只有操作左侧的那个.

  • @ 200OK - 我也不是.如果在其他地方不存在这种行为,那么`__sub__`的行为似乎真的很奇怪所以我去了探险. (3认同)