Python中argsort函数用法

 argsort函数在numpy包中,argsort函数返回的是数组从小到大排序后对应的数组索引

先看一下在一维数组中的表现:

>>> import numpy as np
>>> a = np.array([5,-1,3])
>>> np.argsort(a)
array([1, 2, 0], dtype=int64)

可以看到argsort返回的是对应数组值得索引,同样也是一个数组

在二维及多维数组中也同样可以使用,还可以按行,按列排序返回对应的索引值,还可以降序排列

先建立一个二维数据:

>>> x = np.array([[3,5,1],[6,2,4]])
>>> x
array([[3, 5, 1],
       [6, 2, 4]])

然后分别按行和按列比较大小输出对应的索引,默认是按行比较数值大小输出对应索引值的

>>> np.argsort(x) #默认按行比较
array([[2, 0, 1],
       [1, 2, 0]], dtype=int64)

>>> np.argsort(x,axis = 0) #按列比较
array([[0, 1, 0],
       [1, 0, 1]], dtype=int64)

>>> np.argsort(x,axis = 1) #按行比较
array([[2, 0, 1],
       [1, 2, 0]], dtype=int64)
>>> 

按降序排列

np.argsort(-x)#按降序排列输出索引数组
array([[1, 0, 2],
       [0, 2, 1]], dtype=int64)

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值