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

返回时只是一个排序后对应的原始索引