|
马上注册,结交更多好友,享用更多功能^_^
您需要 登录 才可以下载或查看,没有账号?立即注册
x
找到最大或最小的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处理比较复杂的数据结构
|
|