按列而不是轴对Numpy数组排序

Question

我有一个numpy数组，它等效于：

array([[51, 62, 23],
       [24, 45, 16],
       [67, 78, 39]])

更大。我试图按第二列（即数字62、45、78）排序，但将水平元素保持在一起。我想要得到的输出是：

array([[24, 45, 16],
       [51, 62, 23],
       [67, 78, 39]])

我不知道如何使用np.sort做到这一点-我认为这不是工作的正确功能（尽管我对numpy有点陌生，所以我可能错了）-我我很高兴学习一个新功能，代码片段会很棒，但是一个简单的“ use np.blah”就足够了：D

Answer 1

如果您的numpy数组是arr，则应该可以使用：arr[arr[:,1].argsort()]