cookbook 1.4

大马强

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

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

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

1. num = [-1, -2, 1, 2, 3, 4, 5, 6, 7, 8]
   
2. 
   
3. print(f"前三最小为{heapq.nsmallest(3,num)}")
   
4. print(f"前三最大为{heapq.nlargest(3,num)}")

复制代码

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

二、设置key参数

1. mq = [
   
2.     {"name": "小马", "age": 19},
   
3.     {"name": "小红", "age": 14},
   
4.     {"name": "小白", "age": 15},
   
5.     {"name": "小蓝", "age": 14},
   
6.     {"name": "小率", "age": 12},
   
7.     {"name": "小紫", "age": 11},
   
8.     {"name": "小黄", "age": 10}
   
9. ]
   
10. 
    
11. 
    
12. def MQ(a):
    
13.     return a["age"]
    
14. 
    
15. 
    
16. print(heapq.nsmallest(3, mq, key=lambda s: s["age"]))
    
17. 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处理比较复杂的数据结构