Numpy找到乱序列表中Top-x元素对应值

假设有列表如下

import numpy as np
G =  np.array([6,9,10,34,66,14,71,82,99,11,12,7])
H = np.array([0.1,0.3,0.05,0.08,0.33,0.71,0.91,0.65])
  1. 通过argsort函数进行排序，并返回原列表的索引值
s_G = np.argsort(G)
s_H = np.argsort(H)
  1. 取出对应位置的元素,如取出第二大的元素
print('H列表的第二大元素是：{}'.format(H[s_H[-2]]))
print('G列表的第二大元素是：{}'.format(G[s_G[-2]]))

结果

G列表的第二大元素是：82
H列表的第二大元素是：0.71
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容