【Python】Python3找出List中最大/最小的N个数及索引

Immok

于 2018-04-28 11:37:55 发布

阅读量3.4w

点赞数 5

CC 4.0 BY-SA版权

分类专栏： Python

本文链接：https://blog.csdn.net/ns2250225/article/details/80118621

Python 专栏收录该内容

81 篇文章

订阅专栏

本文介绍了一种使用 Python 的 heapq 模块来找出列表中最大和最小元素及其索引的方法。通过 heapq.nlargest 和 heapq.nsmallest 函数，可以高效地获取指定数量的最大值和最小值，并进一步找到这些数值在原始列表中的位置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

代码

# -*- coding: utf-8 -*-
import heapq

nums = [1, 8, 2, 23, 7, -4, 18, 23, 24, 37, 2]

# 最大的3个数的索引
max_num_index_list = map(nums.index, heapq.nlargest(3, nums))

# 最小的3个数的索引
min_num_index_list = map(nums.index, heapq.nsmallest(3, nums))

print(list(max_num_index_list))
print(list(min_num_index_list))