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

2023-10-05 10:05| 来源: 网络整理| 查看: 265

求取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]

【本文地址】

公司简介

联系我们