集合中最大和最小的项目

为了找到集合中最大的项目，heapq 模块有一个名为 nlargest 的函数，我们传递两个参数，第一个是我们要检索的项目数，第二个是集合名称：

import heapq

numbers = [1, 4, 2, 100, 20, 50, 32, 200, 150, 8]
print(heapq.nlargest(4, numbers))  # [200, 150, 100, 50]

同样，为了找到集合中的最小项，我们使用 nsmallest 函数：

print(heapq.nsmallest(4, numbers))  # [1, 2, 4, 8]

nlargest 和 nsmallest 函数都为复杂的数据结构采用可选参数（关键参数）。以下示例显示了使用 age 属性从 people 字典中检索最老和最年轻的人：

people = [
    {'firstname': 'John', 'lastname': 'Doe', 'age': 30},
    {'firstname': 'Jane', 'lastname': 'Doe', 'age': 25},
    {'firstname': 'Janie', 'lastname': 'Doe', 'age': 10},
    {'firstname': 'Jane', 'lastname': 'Roe', 'age': 22},
    {'firstname': 'Johnny', 'lastname': 'Doe', 'age': 12},
    {'firstname': 'John', 'lastname': 'Roe', 'age': 45}
]

oldest = heapq.nlargest(2, people, key=lambda s: s['age'])
print(oldest)
# Output: [{'firstname': 'John', 'age': 45, 'lastname': 'Roe'}, {'firstname': 'John', 'age': 30, 'lastname': 'Doe'}]

youngest = heapq.nsmallest(2, people, key=lambda s: s['age'])
print(youngest)
# Output: [{'firstname': 'Janie', 'age': 10, 'lastname': 'Doe'}, {'firstname': 'Johnny', 'age': 12, 'lastname': 'Doe'}]