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) 那么排序后的数组 […]