如何从一个集合中找到n个最大或最小的元素?python

import heapq
# heapq模块建立堆数据结构
nums = [1, 8, 2, 3, 123, 43, -1, 0, 90]
# eapq.nlargest(num, list)
lnums = heapq.nlargest(3, nums)
snums = heapq.nsmallest(3, nums)
print(lnums) # [123, 90, 43]
print(snums) # [-1, 0, 1]
numspor = [
    {'name': 'evil woman', 'face': 'unpretty', 'weight': 98},
    {'name': 'A woman', 'face': 'urgly', 'weight': 200},
    {'name': 's woman', 'face': 'beautiful', 'weight': 90},
    {'name': 'new s woman', 'face': 'most beautifu', 'weight': 85}
]
lnunspor = heapq.nlargest(2,numspor, key=lambda x: x['weight'])
snumspor = heapq.nsmallest(2,numspor, key=lambda x: x['weight'])
print(lnunspor)
# [{'name': 'A woman', 'face': 'urgly', 'weight': 200}, {'name': 'evil woman', 'face': 'unpretty', 'weight': 98}]
print(snumspor)
# [{'name': 'new s woman', 'face': 'most beautifu', 'weight': 85}, {'name': 's woman', 'face': 'beautiful', 'weight': 90}]

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值