程序员最近都爱上了这个网站  程序员们快来瞅瞅吧!  it98k网:it98k.com

本站消息

站长简介/公众号

  出租广告位,需要合作请联系站长

+关注
已关注

分类  

暂无分类

标签  

暂无标签

日期归档  

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

发布于2019-10-11 13:30     阅读(1112)     评论(0)     点赞(0)     收藏(4)


求取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)
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
结果:
[94, 90]
[1, 7]
  • 1
  • 2
  • 3

该方法适应性好,即使数值有重复的,也能得到和其数值对应的索引。

方法二:

使用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))
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8

结果:

[94, 90]
{1, 7}
  • 1
  • 2
有重复值的代码:
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)
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10

结果:

[9, 8, 8]
[0, 1, 2]
  • 1
  • 2


所属网站分类: 技术文章 > 博客

作者:加班是一种习惯

链接:https://www.pythonheidong.com/blog/article/134172/4797c6394bf9234143e1/

来源:python黑洞网

任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任

0 0
收藏该文
已收藏

评论内容:(最多支持255个字符)