cookbook 1.4

大马强

找到最大或最小的N个元素

问题描述：
        想在集合中找到最大或最小的N个元素

借助heapq库实现
一、nlargetest() 和 nsmallest

num = [-1, -2, 1, 2, 3, 4, 5, 6, 7, 8]

print(f"前三最小为{heapq.nsmallest(3,num)}")
print(f"前三最大为{heapq.nlargest(3,num)}")

前三最小为[-2, -1, 1]
前三最大为[8, 7, 6]

二、设置key参数

mq = [
    {"name": "小马", "age": 19},
    {"name": "小红", "age": 14},
    {"name": "小白", "age": 15},
    {"name": "小蓝", "age": 14},
    {"name": "小率", "age": 12},
    {"name": "小紫", "age": 11},
    {"name": "小黄", "age": 10}
]


def MQ(a):
    return a["age"]


print(heapq.nsmallest(3, mq, key=lambda s: s["age"]))
print(heapq.nlargest(3, mq, key=MQ))

[{'name': '小黄', 'age': 10}, {'name': '小紫', 'age': 11}, {'name': '小率', 'age': 12}]
[{'name': '小马', 'age': 19}, {'name': '小白', 'age': 15}, {'name': '小红', 'age': 14}]

key参数接受的是一个函数对象,key参数的设置可以让nlargetest() 和 nsmallest处理比较复杂的数据结构