基于最后一列值的numpy排序数组[重复]

Question

import numpy as np

a = np.array([[5,9,44],
              [5,12,43],
              [5,33,11]])

b = np.sort(a,axis=0)

print(b) #not well

# [[ 5  9 11]
# [ 5 12 43]
# [ 5 33 44]]

#desired output:

#[[5,33,11],
# [5,12,43],
# [5,9,44]]

什么 numpy 排序完全改变了行（当然基于从低到高），但我想保持行不变。我想根据最后一列的值对行进行排序，但数组中的行和值必须保持不变。有什么pythonic方法可以做到这一点吗？ 谢谢

Answer 1

ind=np.argsort(a[:,-1])
b=a[ind]

编辑 当您在排序中使用axis 时，它会单独对每一列进行排序，您想要的是从所选列中获取已排序行的索引（-1 相当于最后一列），然后重新排序原始数组。

Answer 2

a[a[:,-1].argsort()] 

可能对你有用