Python程序在字典中查找最高的3个值

有时候我们需要在一个字典中查找最高的几个值，这里提供一种简单直接的方法，使用 Python 标准库中的 heapq 模块。

使用 heapq 模块查找最高的 3 个值

import heapq

# 定义一个字典
d = {'apple': 8, 'banana': 3, 'orange': 5, 'pear': 6, 'watermelon': 1}

# 使用 heapq 模块查找最高的 3 个值
top3 = heapq.nlargest(3, d, key=d.get) # ['apple', 'pear', 'orange']

# 输出结果
print(top3)

代码说明

1. 引入 heapq 模块。
2. 定义一个字典 d。
3. 使用 heapq 模块中的 nlargest() 函数，查找最高的 3 个值。函数的第一个参数指定需要查找的数量，第二个参数传入一个可迭代对象，即字典 d，第三个参数 key 指定按照字典中的值进行比较。
4. 输出结果。

解释说明

heapq 模块是一个堆队列算法模块，它提供了一些方法来在集合中查找最大或最小的 N 个元素。其中 nlargest() 方法就是用来查找最大的 N 个元素，它接收三个参数：

• n：需要查找的最大元素数量。
• iterable：可迭代对象，用于查找目标元素。
• key：指定一个函数，用于告诉 nlargest() 如何从 iterable 中取值进行比较。