Python获取list中最大或最小的n个数及其索引方式
作者：还是少年呀
文章介绍两种获取列表最大/最小值的方法：方法一通过深拷贝逐次定位并置零,适应重复值场景；方法二利用heapq模块,适用于无重复值情况,重复时需结合方法一,两种方法各有优劣,代码示例展示不同场景下的实现效果
求取list中的最大或者最小的几个数字可以使用以下方法
方法一
使用深拷贝的方法，copy需要求索引的list，每次求最大或者最小值及其索引，并置相应位置的值为0，迭代n次。
代码如下：
import copy
m = [34,94,35,78,45,67,23,90,1,0]
t = copy.deepcopy(m)
# 求m个最大的数值及其索引
max_number = []
max_index = []
for _ in range(2):
number = max(t)
index = t.index(number)
t[index] = 0
max_number.append(number)
max_index.append(index)
t = []
print(max_number)
print(max_index)结果: [94, 90] [1, 7]
该方法适应性好，即使数值有重复的，也能得到和其数值对应的索引。
方法二
使用heapq包中的方法，当所求的最大或最小的n个数值不重复时可以直接求出相应的数值和对应的索引，当有重复值的时候还是需要借鉴方法一的思路。
无重复值的代码
import heapq m = [34,94,35,78,45,67,23,90,1,0] # 求一个list中最大的2个数，并排序 max_number = heapq.nlargest(2, m) # 最大的2个数对应的，如果用nsmallest则是求最小的数及其索引 max_index = map(m.index, heapq.nlargest(2, m)) print(max_number) # max_index 直接输出来不是数，使用list()或者set()均可输出 print(set(max_index))
结果：
[94, 90]
{1, 7}
有重复值的代码
import heapq
m = [9, 8, 8, 6, 6, 5, 4, 2, 2, 0]
max_number = heapq.nlargest(3, m)
max_index = []
for t in max_number:
index = m.index(t)
max_index.append(index)
m[index] = 0
print(max_number)
print(max_index)结果:
[9, 8, 8] [0, 1, 2]
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。