结合两个字典词典(Python)

时间:2011-06-06 18:07:01

标签: python dictionary union

有没有一种简单的方法可以在Python中组合两个字典词典?这就是我需要的:

dict1 = {'A' : {'B' : 'C'}}
dict2 = {'A' : {'D' : 'E'}}

result = dict_union(dict1, dict2)
# => result = {'A' : {'B' : 'C', 'D' : 'E'}}

我创建了一个蛮力函数,但我正在寻找一个更紧凑的解决方案:

def dict_union(train, wagon):
    for key, val in wagon.iteritems():
        if not isinstance(val, dict):
            train[key] = val
        else:
            subdict = train.setdefault(key, {})
            dict_union(subdict, val)

6 个答案:

答案 0 :(得分:4)

这是一个实现你正在寻找的行为的类RUDict(用于Recursive-Update dict):

class RUDict(dict):

    def __init__(self, *args, **kw):
        super(RUDict,self).__init__(*args, **kw)

    def update(self, E=None, **F):
        if E is not None:
            if 'keys' in dir(E) and callable(getattr(E, 'keys')):
                for k in E:
                    if k in self:  # existing ...must recurse into both sides
                        self.r_update(k, E)
                    else: # doesn't currently exist, just update
                        self[k] = E[k]
            else:
                for (k, v) in E:
                    self.r_update(k, {k:v})

        for k in F:
            self.r_update(k, {k:F[k]})

    def r_update(self, key, other_dict):
        if isinstance(self[key], dict) and isinstance(other_dict[key], dict):
            od = RUDict(self[key])
            nd = other_dict[key]
            od.update(nd)
            self[key] = od
        else:
            self[key] = other_dict[key]


def test():
    dict1 = {'A' : {'B' : 'C'}}
    dict2 = {'A' : {'D' : 'E'}}

    dx = RUDict(dict1)
    dx.update(dict2)
    print(dx)


if __name__ == '__main__':
    test()


>>> import RUDict
>>> RUDict.test()
{'A': {'B': 'C', 'D': 'E'}}
>>>

答案 1 :(得分:3)

此解决方案非常紧凑。这很难看,但你要求的是一些相当复杂的行为:

dict_union = lambda d1,d2: dict((x,(dict_union(d1.get(x,{}),d2[x]) if
  isinstance(d2.get(x),dict) else d2.get(x,d1.get(x)))) for x in
  set(d1.keys()+d2.keys()))

答案 2 :(得分:1)

我的解决方案旨在按您的方式组合任意数量的词典,并且可以将其限制为仅合并两个词典,从而可以简化整洁的外观,但是其背后的逻辑应该在您的程序中易于使用。

def dictCompressor(*args):
    output = {x:{} for mydict in args for x,_ in mydict.items()}
    for mydict in args:
        for x,y in mydict.items():
            output[x].update(y)
    return output

答案 3 :(得分:0)

您可以继承dict并将原始dict.update()方法包装在一个版本上,该版本会在子区域上调用update()而不是直接覆盖子区域。但是,这可能最终会占用与现有解决方案一样多的努力。

答案 4 :(得分:0)

必须是递归的,因为字典可能会嵌套。这是我第一次接受它,你可能想要在字典嵌套在不同深度时定义你的行为。

def join(A, B):
    if not isinstance(A, dict) or not isinstance(B, dict):
        return A or B
    return dict([(a, join(A.get(a), B.get(a))) for a in set(A.keys()) | set(B.keys())])

def main():
    A = {'A': {'B': 'C'}, 'D': {'X': 'Y'}}
    B = {'A': {'D': 'E'}}
    print join(A, B)

答案 5 :(得分:0)

至于我没有足够的信息,但无论如何请在下面找到我的示例代码:

dict1 = {'A' : {'B' : 'C'}}
dict2 = {'A' : {'D' : 'E'}, 'B':{'C':'D'}}
output = {}
for key in (set(dict1) | set(dict2):
    output[key] = {}
    (key in dict1 and output[key].update(dict1.get(key)))
    (key in dict2 and output[key].update(dict2.get(key)))