用于多维ndarray的argsort

用于多维ndarray的argsort

上面的答案现在已经过时了，因为在numpy
1.15中添加了新功能以使其更简单。take_along_axis（https://docs.scipy.org/doc/numpy-1.15.1/reference/generation/numpy.take_along_axis.html）允许您执行以下 *** 作：

>>> a = np.array([[3,1,2],[8,9,2]])>>> np.take_along_axis(a, a.argsort(axis=-1), axis=-1)array([[1 2 3]       [2 8 9]])