Numpy中的argsort函数详解

时间:2021-09-11 21:25:27

Numpy中的argsort函数返回的是每个元素的排序序号,但是不是很容易理解:

>>> dd=mat([4,5,1])
>>> dd.argsort()
matrix([[2, 0, 1]])

一开始的时候想不明白为什么是2,0,1而不是1,2,0.因为安从小到大排序[1,4,5],1的位置是0,4的位置是1,5的位置是2。所以[4,5,1]的位置是[1,2,0],这种理解是错误的。正确的理解应该是[4,5,1]中4的索引是0,5的索引是1,1的索引是2,按[1,4,5]的顺序应该是[2,0,1]

相关文章