np.argsort的理解

今天在学习的时候用到了numpy库的一个argsort函数,感觉是每个值的大小排序,例如使用

np.argsort([5,4,10,7])

打印出来的结果为:array([1, 0, 3, 2], dtype=int64)
嗯不错感觉,4是最小的,所以第二个值为0,10是最大的所以第三个值为3,好像是这么回事,但是我又进一步实验了一下,如下面这个例子

print(np.argsort([5,4,1,3]))

那么按照我一开始的想法打印的结果应该是[3,2,0,1],但是我们可以自己实验一下,会发现结果是:
array([2, 3, 1, 0], dtype=int64)这并不是我们预测的结果啊。后来仔细查看了一下,才发现打印出来的是这样的:列表中第一个元素值是[5,4,1,3]这个列表最小值所在位置的索引,比如1是最小值,在列表中的索引下标是2,所以第一个位置的值为2,后面的依次查找就对了。
在二维中默认是按照行进行比较的,例如

np.argsort([[5,4,1,3],[4,6,8,7]])

输出结果为
array([[2, 3, 1, 0],
[0, 1, 3, 2]], dtype=int64)
要想按照每一列进行查看,就需要指定参数axis=0

np.argsort([[5,4,1,3],[4,6,8,7]],axis=0)

结果为
array([[1, 0, 0, 0],
[0, 1, 1, 1]], dtype=int64)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值