python 求列表中最大/最小n个元素的索引_python取列表最小的n个元素的索引

作者：繁依Fanyi0 | 2024-03-07 09:14:24

踩

python取列表最小的n个元素的索引

方法：首先利用heapq模块中的nlargest/nsmallest方法，求得列表中的前n个最大/最小元素，然后利用index（）求每个元素的索引。

代码如下：

import heapq

l = [5,6,5,3,2,7,8]

max_num = heapq.nlargest(4,l)

max_index = []
for i in max_num:
    max_index.append(l.index(i))
print(max_index)

#输出：
#[6, 5, 1, 0]
1
2
3
4
5
6
7
8
9
10
11
12
13

借助python的map()方法优化代码：

import heapq

l = [5,6,5,3,2,7,8]

max_index_new = list(map(l.index, heapq.nlargest(4,l)))
print(max_index_new)

#输出：
#[6, 5, 1, 0]
1
2
3
4
5
6
7
8
9

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/204699?site