基于自定义函数按键合并字典值

时间:2015-03-28 09:30:22

标签: python python-2.7

假设您有两个词典,并且希望通过将函数应用于具有匹配键的值来合并两个词典。这里我使用+运算符作为二元函数。

x = { 1: "a", 2: "b", 3: "c" }
y = { 1: "A", 2: "B", 3: "C" }

result = { t[0][0]: t[0][1] + t[1][1] for t in zip(sorted(x.items()), sorted(y.items())) }

print result # gives { 1: "aA", 2: "bB", 3: "cC" }

我更喜欢自包含的表达式而不是语句,但这是不可读的。

到目前为止我正在做:

def dzip(f, a, b):
    least_keys = set.intersection(set(a.keys()), set(b.keys()))
    copy_dict = dict()
    for i in least_keys.keys():
        copy_dict[i] = f(a[i], b[i])
    return copy_dict

print dzip(lambda a,b: a+b,x,y)

有没有比我给出的表达式更可读的解决方案?

2 个答案:

答案 0 :(得分:2)

在第一种情况下,您可以直接使用词典理解:

>>> x = { 1: "a", 2: "b", 3: "c" }
>>> y = { 1: "A", 2: "B", 3: "C" }
>>> {key: x.get(key, "") + y.get(key, "") for key in set.intersection(set(x.keys()), set(y.keys()))}
{1: 'aA', 2: 'bB', 3: 'cC'}

因此,在第二段代码中,您可以将其简化为简单的一行代码:

def dzip(f, a, b):
    return {key: f(a.get(key, ""), b.get(key, "")) for key in set.inersection(set(a.keys()) + set(b.keys()))}

您甚至可以将dzip定义为lambda:

dzip = lambda f, a, b: {key: f(a.get(key, ""), b.get(key, "")) 
    for key in set.intersection(set(a.keys()), set(b.keys()))}

在一次运行中,这变为:

>>> dzip = lambda f, a, b: {key: f(a.get(key, ""), b.get(key, "")) 
...         for key in set.intersection(set(a.keys()), set(b.keys()))}    
>>> 
>>> print dzip(lambda a,b: a+b,x,y)
{1: 'aA', 2: 'bB', 3: 'cC'}

请注意,即使x和y具有不同的键集(只是在您的第一个代码版本中可能会中断的内容),这也会起作用。

答案 1 :(得分:1)

您可以将Counter用于此类dict合并

from collections import Counter
>>>Counter(x)+Counter(y)
Counter({3: 'cC', 2: 'bB', 1: 'aA'})