比较嵌套字典

Question

我想比较嵌套字典如下：

d = {'siteA': {'00000000': 3, '11111111': 4, '22222222': 5},
        'siteB': {'00000000': 1, '11111111': 2, '22222222': 5}}

e = {'siteA': {'00000000': 5}}

f = {'siteB': {'33333333': 10}}

g = {'siteC': {'00000000': 8}}

d 是用于与e、f 和g 进行比较的完整字典总数。

如果 e 恰好在 siteA-00000000 中找到，那么我希望两个值（在本例中为 3 和 5）加起来为 8。

如果找不到f（在这种情况下是这样），我想将字典附加到d['siteB']。

如果找不到g，想追加到d。

谢谢！

Answer 1

collections.Counter 可用于对字典中的值求和并在它们不存在的地方添加键。由于Counter 是dict 的子类，因此不应破坏其他操作。除了一次性的转换成本外，它还非常高效且专为此类任务而设计。

from collections import Counter

# convert d to dictionary of Counter objects
d = {k: Counter(v) for k, v in d.items()}

# add items from e
for k, v in e.items():
    if k in d:
        d[k] += Counter(e[k])

# add item from f if not found
for k, v in f.items():
    if k not in d:
        d[k] += Counter(f[k])

# add item from g if not found
for k, v in g.items():
    if k not in d:
        d[k] = Counter(v)

结果：

print(d)

{'siteA': Counter({'00000000': 8, '11111111': 4, '22222222': 5}),
 'siteB': Counter({'00000000': 1, '11111111': 2, '22222222': 5}),
 'siteC': Counter({'00000000': 8})}

Answer 2

您可以将collections 中的Counter 与defaultdict 结合使用。

顾名思义，计数器计数相同的元素，而 defaultdict 允许您通过提供默认值（在本例中为空的 Counter）来访问不存在的键。然后你的代码就变成了

from collections import Counter, defaultdict


d = defaultdict(Counter)
d['siteA'] = Counter({'00000000': 3, '11111111': 4, '22222222': 5})
d['siteB'] = Counter({'00000000': 1, '11111111': 2, '22222222': 5})
print(d.items())

> dict_items([('siteA', Counter({'22222222': 5, '11111111': 4, '00000000': 3})), 
>             ('siteB', Counter({'22222222': 5, '11111111': 2, '00000000': 1}))])

# d + e:
d['siteA'].update({'00000000': 5})
print(d.items())

> dict_items([('siteA', Counter({'00000000': 8, '22222222': 5, '11111111': 4})), 
>             ('siteB', Counter({'22222222': 5, '11111111': 2, '00000000': 1}))])

# d + f
d['siteB'].update({'33333333': 10})
print(d.items())

> dict_items([('siteA', Counter({'00000000': 8, '22222222': 5, '11111111': 4})), 
>             ('siteB', Counter({'33333333': 10, '22222222': 5, '11111111': 2, '00000000': 1}))])

# d + g
d['siteC'].update({'00000000': 8})
print(d.items())

> dict_items([('siteA', Counter({'00000000': 8, '22222222': 5, '11111111': 4})), 
>             ('siteB', Counter({'33333333': 10, '22222222': 5, '11111111': 2, '00000000': 1})), 
>.            ('siteC', Counter({'00000000': 8}))])

Answer 3

鉴于您的字典 dict[site][address] 的格式，假设这个合并函数将从 dictFrom 中获取值并根据您的规则将它们插入到 dictTo 中。

def merge(dictTo, dictFrom):
    for site in dictFrom:
        if site not in dictTo:
            dictTo[site] = {}

        for address in dictFrom[site]:
            dictTo[site][address] = dictTo[site].get(address, 0) + dictFrom[site][address]

merge(d, e)
merge(d, f)
merge(d, g)

这可能比 jpp 的答案更可取，因为 dict[site] 上的对象仍然是基本的 dicts。