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

最新推荐文章于 2026-05-16 06:09:24 发布

原创最新推荐文章于 2026-05-16 06:09:24 发布 · 3.4w 阅读

25 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

Python

当前文章被收录于：

Python

81 篇文章 2 人学习

订阅专栏查看详情

当前文章被以下社区和专栏收录：

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

代码

# -*- 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))