找到最大或最小的N个值

对于python原生的数据类型来说,并不存在直接的方法可以找到最大或最小的N个值,

传统的方法必须先排序,然后再截取相应的值,而且对于集合这类数据类型来说还不能直接排序,

需要先转化为列表才行,有的时候非常不便,因此python提供了内置模块heapq可以解决这类问题。

下面分别对列表、集合和字典的角度解决这个需求。

1.列表

#列表
nums = [45,12,4,-5.34,23,76,85,-15,56]

# (1)传统方法
nums2 = nums[:]
nums2.sort()
print(nums2[-3:]) #[56, 76, 85]

# (2)引入库heapq
#取最大的三个值
print(heapq.nlargest(3,nums))
#结果[85, 76, 56]
print(heapq.nsmallest(3,nums)) #[-15, -5.34, 4]

2.集合

#集合
set1 = {45,12,4,-5,34,23,76,85,-15,56}
# (1)传统方法
#由于集合不能排序,因此需要先将集合转化为列表
list1 = list(set1)
list1.sort()
print(list1[-3:])  #[56, 76, 85]

#(2)更时尚的方法
print(heapq.nlargest(3,set1))  #[85, 76, 56]
print(heapq.nsmallest(3,set1))  #[-15, -5, 4]

3.字典

#字典
names = [
    {"name":"科比","age":26},
    {"name":"大姚","age":38},
    {"name":"丁同","age":19},
    {"name":"志成","age":11},
    {"name":"拉塞尔","age":84},
]
# (1)第一种方法heapq
#找出年龄最大的两位(专门取值的方法)
print(heapq.nlargest(2,names,lambda x:x["age"]))

# (2)第二种(传统方法,可以排序,可以取值)
#可以给排序指定一个字典中的键
names.sort(key=lambda x:x["age"])
print(names)

# (3)第三种
# 通过公共键对字典排序然后取值,
from operator import itemgetter
new_names = sorted(names,key=itemgetter("age"))
print(new_names[-2:])
# 第三种方法是第二种方法的浓缩,不过如果你知道第三种为何和压迫使用第二种了。

对于字典还可以使用zip反转然后排序

name={"科比":26,"大姚":38,"丁同":19,"拉塞尔":84,"志成":11}
print(sorted(zip(name.values(),name.keys())))
[(11, '志成'), (19, '丁同'), (26, '科比'), (38, '大姚'), (84, '拉塞尔')]
原文地址:https://www.cnblogs.com/yangmingxianshen/p/8331808.html