在字典中求多重最小值

2024-04-20 11:59:27 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个关于dict的python dict,我想找到每个内部dict的最小值,对于外部dict中的每个键。在

为了给出一个简化的版本,假设我们有这个dicts的dict:

my_dict = {'a': {'b': 10, 'c': 5, 'd': 6, 'e': 6, 'f': 6}, 'b': {'a': 10, 'c': 5, 'd': 4, 'e': 4, 'f': 4}, 'c': {'a': 5, 'b': 5, 'd': 7, 'e': 7, 'f': 7}, 'd': {'a': 6, 'b': 4, 'c': 7, 'e': 0, 'f': 0},'e': {'a': 6, 'b': 4, 'c': 7, 'd' : 0, 'f': 0}, 'f': {'a': 6, 'b': 4, 'c': 7, 'd': 0, 'e': 0}}

我想创建一个新的dict,其中对于每个内部dict,我找到该dict的最小值并获取相应的键值。在

因此,对于带有dict {'b': 10, 'c': 5, 'd': 6, 'e': 6, 'f': 6}的键a,最小值是5,因此我将把a放入分配给空列表的新dict中,然后将c追加到列表中,因为它对应于值5。在

上述列表的最终输出如下所示:

^{pr2}$

我遇到的问题是如果存在多个最小值。例如,f的内部dict有两个值为0的键(d和{}),因此都应该返回。在

以下是我目前掌握的代码:

new_dict = {}
for k, v in my_dict.items():
    new_dict[k] = [min(v, key=v.get)]

print((new_dict))

这给了我:

{'a': ['c'], 'b': ['d'], 'c': ['a'], 'd': ['e'], 'e': ['d'], 'f': ['d']}

所以它返回正确的最小值,只是不是所有的最小值。在

谢谢你的时间。在


Tags: key代码in版本列表newforget
1条回答
网友
1楼 · 发布于 2024-04-20 11:59:27

您可以在for循环中进行小的修改,并在生成键列表之前从内部字典的值中获取最小值:

new_dict = {}
for k, v in my_dict.items():
    min_val = min(v.values())
    new_dict[k] = [sk for sk, sv in v.items() if sv == min_val]

print(new_dict)
# {'a': ['c'], 'c': ['a', 'b'], 'f': ['e', 'd'], 'e': ['f', 'd'], 'd': ['f', 'e'], 'b': ['e', 'd', 'f']}

相关问题 更多 >