我有两个字典,我希望将它们合并成一个字典,使得合并后的字典中每个键对应的值都是在原始两个字典中该键对应的值中的最小值。
h1 = {"a":3, "b":5, "c":2}
h2 = {"a":1, "c":5, "d":10}
result = merge(h1, h2) = {"a":1, "b":5, "c":2, "d":10}
有没有一个很酷的一行代码可以实现这个功能?如果没有,那么最优雅的方法是什么?
我有两个字典,我希望将它们合并成一个字典,使得合并后的字典中每个键对应的值都是在原始两个字典中该键对应的值中的最小值。
h1 = {"a":3, "b":5, "c":2}
h2 = {"a":1, "c":5, "d":10}
result = merge(h1, h2) = {"a":1, "b":5, "c":2, "d":10}
您可以这样做
>>> {k: min(i for i in (h1.get(k), h2.get(k)) if i) for k in h1.viewkeys() | h2}
{'a': 1, 'c': 2, 'b': 5, 'd': 10}
h1.viewkeys() | h2
实际上是找到了集合的并集,并获取所有在 h1
或者 h2
中的键。然后,我们找到了 `h1` 和 `h2` 中相应键的最小值。
如果你正在使用 Python 3.x,那么你只需要使用 keys
,像这样:
>>> {k : min(i for i in (h1.get(k), h2.get(k)) if i is not None) for k in h1.keys() | h2}
{'d': 10, 'a': 1, 'b': 5, 'c': 2}
你也可以试试这个:
>>> {k: min(h1.get(k) or h2[k], h2.get(k) or h1[k]) for k in (h1.keys() + h2.keys())}
{'a': 1, 'c': 2, 'b': 5, 'd': 10}
h1.keys() + h2.keys()
。 - styvane
if i
替换为if i is not None
可以允许 0 值,这一点反映在编辑中。 - dcsuka