numpy.argsort返回值的理解

函数声明

numpy.argsort(a, axis=-1, kind='quicksort', order=None)

功能说明

Returns the indices that would sort an array.
返回数组排序的索引.

问题说明

实际使用过程中,会对返回值的解释产生疑问.
如manual中给的示例:

x = np.array([3, 1, 2])
np.argsort(x)

运行结果:
array([1, 2, 0])

整理为表格:

元素1 元素2 元素3
数组元素 3 1 2
返回值(索引) 1 2 0

像我这样第一次接触的人就误认为 [3,1,2]对应的排序是(1,2,0),也就是排序后的数组为2,3,1.
很显然这种理解是错误的.

正确理解

正确的理解也很简单,毕竟是通用库里的函数,不会搞的很复杂.
正确的理解是返回值返回的是原数组元素的索引,如上例,[3,1,2]对应的索引分别是(0,1,2).换一种表达方式,把索引用带有括号的形式表示:

元素 索引
3 (0)
1 (1)
2 (2)

那么上例中返回值记为:(1), (2), (0)
那么排序后的数组为: 1, 2, 3

参考

[1] argsort文档


原创文章,转载请注明: 转载自张哲的博客

本文链接地址: numpy.argsort返回值的理解