python - 如何合并两个字典以保持公共键的最大值？

Question

我有两本看起来像的字典：

{'r': 2, 'e': 4, 'h': 2, 'k': 4}

和

{'r': 2, 'e': 5, 'y': 2, 'h': 2}

我如何获得一个包含所有键的字典，但如果两个初始字典中都有键，它会为该键保留更高的值？我想要一个看起来像这样的字典：

{'e': 5, 'k': 4, 'y': 2, 'h': 2, 'r': 2}

以前的答案都没有帮助我。

Answer 1

您可以使用itertools.chain组合所有值，然后itertools.groupby获取每个单独键的所有值，然后取这些值的最大值。您需要在使用 groupby 之前对合并的数据进行排序，以便它正常工作。我也operator.itemgetter用来获取键和值而不是 lambdas，所以如果你不想导入另一个库，你可以用 lambdas 替换它们，尽管我不建议这样做，因为它更慢并且不需要使用它们真的。

from itertools import chain, groupby
from operator import itemgetter

data1 = {'r': 2, 'e': 4, 'h': 2, 'k': 4}
data2 = {'r': 2, 'e': 5, 'y': 2, 'h': 2}

get_key, get_val = itemgetter(0), itemgetter(1)
merged_data = sorted(chain(data1.items(), data2.items()), key=get_key)

output = {k: max(map(get_val, g)) for k, g in groupby(merged_data, key=get_key)}

print(output)

---

{'e': 5, 'h': 2, 'k': 4, 'r': 2, 'y': 2}

这里的另一种选择是collections.defaultdict，如果有负值float('-inf')用作默认值，请确保您始终获得正确的输出：

from collections import defaultdict

output = defaultdict(lambda: float('-inf'))

for d in (data1, data2):
    for k, v in d.items():
        output[k] = max(output[k], v)

print(dict(output))

---

{'r': 2, 'e': 5, 'h': 2, 'k': 4, 'y': 2}

或者没有任何进口dict.setdefault基本上可以代替defaultdict：

output = {}

for d in (data1, data2):
    for k, v in d.items():
        output.setdefault(k, float('-inf'))
        output[k] = max(output[k], v)
        
print(output)

---

{'r': 2, 'e': 5, 'h': 2, 'k': 4, 'y': 2}

最后，使用pandas

import pandas as pd

data1 = {'r': 2, 'e': 4, 'h': 2, 'k': 4}
data2 = {'r': 2, 'e': 5, 'y': 2, 'h': 2}
    
res = pd.concat(map(pd.DataFrame, ([data1], [data2]))).max().astype(int).to_dict()

Answer 2

您可以创建两个 dict 键的联合集，然后使用一个 dict 理解，它为每个键获取最大值：

keys = set(a.keys()).union(b.keys())
output = {k:max(a.get(k,float('-inf')), b.get(k, float('-inf'))) for k in keys}