行中的numpy&#argsort工作效果不如预期

时间:2016-10-05 02:33:40

标签: python arrays numpy

我有一个简单的python代码段,用于对2D numpy矩阵的每一行进行排序

>>> import numpy as np
>>> X=np.random.random((3,3))
>>> X
array([[ 0.57758986,  0.76771506,  0.94539164],
       [ 0.37619117,  0.85602478,  0.21527702],
       [ 0.41134957,  0.47028597,  0.0672778 ]])
>>> X.argsort(axis=-1)
array([[0, 1, 2],
       [2, 0, 1],
       [2, 0, 1]], dtype=int64)

我要做的是对每一行进行排序,以便我可以从最小到最大得到每行的顺序,所以我期望X.argsort的输出为:

array([[0, 1, 2],
       [1, 2, 0],
       [1, 2, 0]], dtype=int64)

然而绝对argsort不会以这种方式运作,我对numpy的argsort有什么误解吗?

感谢您的任何建议!

0 个答案:

没有答案