Python获取list中最大或最小的n个数及其索引

求取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]