Python 求复杂序列中最大、最小N个元素

目录

• 一、问题
• 二、解决问题
• 三、讨论

一、问题

求复杂序列中的最大、最小N个元素。

二、解决问题

heapq 模块有两个函数：nsmallest、nlargest()。

import heapq    # 堆排序

nums = [1, 8, 2, 23, 7, -4, 18, 23, 42, 37, 2]
print(heapq.nlargest(3, nums))
print(heapq.nsmallest(3, nums))

输出：

[42, 37, 23]
[-4, 1, 2]

复杂序列：

import heapq

info = [
    {'name': 'wangke', 'age': 30, 'weight': 65},
    {'name': 'wangyan', 'age': 28, 'weight': 85},
    {'name': 'wangying', 'age': 28, 'weight': 75},
    {'name': 'qinlu', 'age': 18, 'weight': 50},
    {'name': 'hongxia', 'age': 56, 'weight': 60}
]

# 输出年龄最小的3条数据
age = heapq.nsmallest(3, info, key=lambda s: s['age'])
print(age)

输出：

[{'name': 'qinlu', 'age': 18, 'weight': 50}, 
 {'name': 'wangyan', 'age': 28, 'weight': 85}, 
 {'name': 'wangying', 'age': 28, 'weight': 75}]

三、讨论

查找的元素个数相对比较小，用 nsmallest()、nlargest()。

查找唯一的最大、最小值，用 max()、min()。

查找元素个数与总的元素个数接近，通常先排序，然后再切片。sorted()。