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 itertools模块中starmap函数的应用
starmap是一个非常有用的函数,它属于itertools模块中的一部分,本文将详细介绍starmap函数的作用、用法以及实际应用场景,希望对大家有所帮助2024-03-03
python3中celery异步框架简单使用+守护进程方式启动
这篇文章主要介绍了python3中celery异步框架简单使用+守护进程方式启动,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2021-01-01
python使用PIL和matplotlib获取图片像素点并合并解析
这篇文章主要介绍了python使用PIL和matplotlib获取图片像素点并合并解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-09-09
PyTorch基于Transformer架构的完整文本生成实现方案
PyTorch文本生成代码模板与解析 本文提供了一个基于Transformer架构的完整文本生成实现方案,包含以下核心内容: 代码架构: 完整实现从数据预处理到模型训练的端到端流程 包含Transformer核心组件,需要的朋友可以参考下2026-04-04


最新评论