假设我有一个 NumPy 数组:
[[4 9 2]
[5 1 3]]
我想对这个数组的底行进行排序,但让顶行跟随排序,这样我得到:
[[9 2 4]
[1 3 5]]
我知道您可以使用 sorted() 函数进行排序,但这需要列表的输入和输出。
有什么想法吗?非常感谢!
最佳答案
import numpy as np
a = np.array([[4,9,2],[5,1,3]])
idx = np.argsort(a[1])
现在您可以使用 idx 来索引您的数组:
b=a[:,idx]
关于python - 使用 NumPy 进行二维排序 - 对 1 行进行排序,让另一行跟随排序,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/7900171/