Python sort np.argsort()元素从小到大排序后,提取对应的索引index,可以一行搞定排序
https://www.cnblogs.com/cgmcoding/p/13590505.html
np.argsort()元素从小到大排序后,提取对应的索引index,可以一行搞定排序
函数用法
numpy.argsort(a, axis=-1, kind=’quicksort’, order=None)
功能: 将矩阵a按照axis排序,并返回排序后的下标,axis=0是列,1是行
参数: a:输入矩阵, axis:需要排序的维度
返回值: 输出排序后的下标
import numpy as np x = np.array([1,4,3,-1,6,9]) x.argsort() # array([3, 0, 1, 2, 4, 5], dtype=int64)
可以发现,argsort()是将X中的元素从小到大排序后,提取对应的原来的索引index,然后输出到y
#所以取数组x的最小值的索引位置可以写成: x[x.argsort()[0]] #或者用argmin()函数 x[x.argmin()] #数组x的最大值,写成: x[x.argsort()[-1]] # -1代表从后往前反向的索引 #或者用argmax()函数,不再详述 x[x.argmax()]
输出排序后的数组
x[x.argsort()] #或者 x[np.argsort(x)] #array([-1, 1, 3, 4, 6, 9])
那么给数据排序不就很简单了,直接一行就搞定了
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· 【自荐】一款简洁、开源的在线白板工具 Drawnix