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

 更新时间:2025年09月01日 09:51:43   作者:还是少年呀  
文章介绍两种获取列表最大/最小值的方法:方法一通过深拷贝逐次定位并置零,适应重复值场景;方法二利用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]

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • Python轻松实现提取视频音频并去除静音片段

    Python轻松实现提取视频音频并去除静音片段

    在处理视频素材时,我们常常需要提取其中的音频,并且希望去除冗长的静音部分,借助Python的几个优秀库,我们可以快速实现这一需求,下面我们就来看看具体实现方法吧
    2025-10-10
  • 详解Python itertools模块中starmap函数的应用

    详解Python itertools模块中starmap函数的应用

    starmap是一个非常有用的函数,它属于itertools模块中的一部分,本文将详细介绍starmap函数的作用、用法以及实际应用场景,希望对大家有所帮助
    2024-03-03
  • python3中celery异步框架简单使用+守护进程方式启动

    python3中celery异步框架简单使用+守护进程方式启动

    这篇文章主要介绍了python3中celery异步框架简单使用+守护进程方式启动,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2021-01-01
  • Python简易计算器制作方法代码详解

    Python简易计算器制作方法代码详解

    这篇文章主要介绍了Python简易计算器制作方法,文中代码主要用到了python中的tkinter库,代码简单易懂,非常不错,具有一定的参考借鉴价值,需要的朋友可以参考下
    2019-10-10
  • python使用PIL和matplotlib获取图片像素点并合并解析

    python使用PIL和matplotlib获取图片像素点并合并解析

    这篇文章主要介绍了python使用PIL和matplotlib获取图片像素点并合并解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-09-09
  • 对于Python的Django框架部署的一些建议

    对于Python的Django框架部署的一些建议

    这篇文章主要介绍了对于Python的Django框架部署的一些建议,包括项目文件的布局等,需要的朋友可以参考下
    2015-04-04
  • win10 64bit下python NLTK安装教程

    win10 64bit下python NLTK安装教程

    这篇文章主要为大家详细介绍了win10 64bit下python NLTK安装教程,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2018-09-09
  • Python softmax实现及数值稳定性详解

    Python softmax实现及数值稳定性详解

    这篇文章主要为大家介绍了Python softmax实现及数值稳定性详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-07-07
  • Python实现图片背景移除工具

    Python实现图片背景移除工具

    这篇文章主要为大家详细介绍了如何通过Python语言实现一个简单的图片背景移除工具,文中的示例代码讲解详细,感兴趣的小伙伴可以了解一下
    2023-02-02
  • PyTorch基于Transformer架构的完整文本生成实现方案

    PyTorch基于Transformer架构的完整文本生成实现方案

    PyTorch文本生成代码模板与解析 本文提供了一个基于Transformer架构的完整文本生成实现方案,包含以下核心内容: 代码架构: 完整实现从数据预处理到模型训练的端到端流程 包含Transformer核心组件,需要的朋友可以参考下
    2026-04-04

最新评论