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]
前三最大为
二、设置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处理比较复杂的数据结构
页:
[1]