9

问题：

>>> a = dict(a=1,b=2    )
>>> b = dict(    b=3,c=2)

>>> c = ???

c = {'a': 1, 'b': 5, 'c': 2}

因此，这个想法是两个以最短形式通过 int/float 值添加到字典中。这是我设计的一种解决方案，但我不喜欢它，因为它很长：

c = dict([(i,a.get(i,0) + b.get(i,0)) for i in set(a.keys()+b.keys())])

我认为必须有一个更短/简洁的解决方案（可能与 reduce 和 operator 模块有关？itertools？）......有什么想法吗？

更新：我真的希望找到更优雅的东西，比如“reduce(operator.add, key = itemgetter(0), a+b)”。（显然这不是真正的代码，但你应该明白）。但这似乎是一场梦。

更新：仍在寻找更简洁的解决方案。也许 groupby 可以提供帮助？我提出的使用“reduce”/“groupby”的解决方案实际上并不简洁：

from itertools import groupby
from operator import itemgetter,add

c = dict( [(i,reduce(add,map(itemgetter(1), v))) \
              for i,v in groupby(sorted(a.items()+b.items()), itemgetter(0))] )
4

8 回答 8

9

最容易使用Counter

>>> from collections import Counter
>>> a = dict(a=1,b=2    )
>>> b = dict(    b=3,c=2)
>>> Counter(a)+Counter(b)
Counter({'b': 5, 'c': 2, 'a': 1})
>>> dict(Counter({'b': 5, 'c': 2, 'a': 1}))
{'a': 1, 'c': 2, 'b': 5}
于 2012-08-07T00:42:22.160 回答
8

solving not in terms of "length" but performance, I'd do the following:

>>> from collections import defaultdict
>>> def d_sum(a, b):
        d = defaultdict(int, a)
        for k, v in b.items():
            d[k] += v
        return dict(d)
>>> a = {'a': 1, 'b': 2}
>>> b = {'c': 2, 'b': 3}
>>> d_sum(a, b)
{'a': 1, 'c': 2, 'b': 5}

it's also py3k-compatible, unlike your original code.

于 2009-05-18T15:00:51.957 回答
6

在我的第一印象中，我会写：

>>> c = a.copy()
>>> for k in b: c[k] = c.get(k, 0) + b[k]
于 2009-05-18T11:58:41.077 回答
5

如果你想要短代码，你就在那里。

如果您想要干净的代码，请继承 Ber'sdefaultdict和重载__add__：

from collections import defaultdict

class summable(defaultdict):
    def __add__(self, rhs):
        new = summable()
        for i in (self.keys() + rhs.keys()):
            new[i] = self.get(i, 0) + rhs.get(i, 0)
        return new

a = summable(int, a=1, b=2)
b = summable(int, b=3, c=4)
c = a + b
print c

给出：

>>> 
defaultdict(None, {'a': 1, 'c': 4, 'b': 5})
>>>
于 2009-05-18T12:05:28.720 回答
4

我认为一行代码已经很短了:)

如果您使用defaultdict并删除一些不必要的列表并设置创建，我可能会变成“半行” ：

from collections import defaultdict

a = defaultdict(int, a=1, b=2)
b = defaultdict(int, b=3, c=4)

c = dict((k, a[k]+b[k]) for k in (a.keys() + b.keys()))
print c
于 2009-05-18T11:55:33.483 回答
3

我想到的第一件事是效率更高，（IMO）更优雅，但仍然打字太多。实际上，它大约相当于kcwu的。

c = reduce(lambda(d, k): [d.update({k: d.get(k, 0) + b[k]}), d][1], b, a.copy())

dict.update不回来真是可惜了self。我想这不是 Python 的方式。如果是这样，这个[..., d][1]把戏就没有必要了。

Perl：“易事易，难事皆有可能”

%a = (a => 1, b => 2);
%b = (b => 3, c => 2);

%c = (%a, map {$_ => $a{$_} + $b{$_}} keys %b);

Haskell：“易事难，难事易” “难事易事，不可能的事刚刚发生”

import qualified Data.Map as M

a = M.fromList [('a', 1), ('b', 2)]
b = M.fromList [('b', 3), ('c', 2)]

c = M.unionWith (+) a b
于 2009-06-17T20:44:17.127 回答
2

评论@John Pirie 的回答：

这是不使用的实现(self.keys() + rhs.keys())：

from collections import defaultdict

class sumdict(defaultdict):
    def __add__(self, rhs):
        d = self.copy() 
        d += rhs
        return d
    __radd__ = lambda self, lhs: self + lhs
    def __iadd__(self, rhs):
        for k, v in rhs.items():
            self[k] += v
        return self

a = sumdict(int, a=1, b=2)
b = dict(b=3, c=4)
c = b + a
a += b

assert a == c == {'a': 1, 'c': 4, 'b': 5} != b
于 2009-06-22T13:37:11.930 回答
1 
def GenerateSum():
  for k in set(a).union(b):
    yield k, a.get(k, 0) + b.get(k, 0)

e = dict(GenerateSum())
print e

或者，使用一个衬垫：

 print dict((k, a.get(k,0) + b.get(k,0)) for k in set(a).union(b))
于 2009-05-18T17:05:09.803 回答