鱼C论坛

 找回密码
 立即注册
查看: 740|回复: 0

[技术交流] cookbook 1.4

[复制链接]
发表于 2022-1-17 09:16:15 | 显示全部楼层 |阅读模式

马上注册,结交更多好友,享用更多功能^_^

您需要 登录 才可以下载或查看,没有账号?立即注册

x

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

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

借助heapq库实现
一、nlargetest() 和 nsmallest
  1. num = [-1, -2, 1, 2, 3, 4, 5, 6, 7, 8]

  2. print(f"前三最小为{heapq.nsmallest(3,num)}")
  3. 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. def MQ(a):
  11.     return a["age"]


  12. print(heapq.nsmallest(3, mq, key=lambda s: s["age"]))
  13. 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处理比较复杂的数据结构

本帖被以下淘专辑推荐:

想知道小甲鱼最近在做啥?请访问 -> ilovefishc.com
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

小黑屋|手机版|Archiver|鱼C工作室 ( 粤ICP备18085999号-1 | 粤公网安备 44051102000585号)

GMT+8, 2024-3-29 15:24

Powered by Discuz! X3.4

© 2001-2023 Discuz! Team.

快速回复 返回顶部 返回列表