使用numpy的argsort函数对二维数组按行（列）排序，返回排序索引时出错

Question

问题遇到的现象和发生背景

版本：numpy 1.22.3 。使用numpy的argsort函数对二维数组按行（列）排序，返回排序索引时出错

问题相关代码，请勿粘贴截图

>>> array = [[0, 1, 2, 3, 4, 5], 
         [444, 4, 8, 3, 1, 10], 
         [2, 5, 8, 999, 1, 4]]
>>> np.argsort(array, axis=0)
array([[0, 0, 0, 0, 1, 2],
       [2, 1, 1, 1, 2, 0],
       [1, 2, 2, 2, 0, 1]], dtype=int64)

运行结果及报错内容

没有报错，但是，第五、六列按列排序的索引有很明显的问题

我的解答思路和尝试过的方法

我的思路，以第六列为例：

我尝试过转置之后一行一行地执行再拼接，但仍然有问题

我想要达到的结果

返回正确的排序索引

Answer 1

有一种pandas的方法可以实现（你问的应该是rank）,需要先转换为dataframe，再用rank解决：
arr=np.array(pd.DataFrame(array).rank(axis=0,method='min'))
这是从1-n的排序，如果想从0开始，后面直接减去一即可
目前我还没有搜查到numpy有直接的方法，需要再查一下