Python Numpy 按指定行或列排序

本文使用的方法是参考自下面链接，感谢！
Sorting arrays in NumPy by column

arr = np.array([[1,3,5],[4,2,0]])

print(f'arr\n{arr}')

# 按第2列排序（column index = 1）
print('按第2列排序\n',arr[arr[:,1].argsort()])

# 按第2行排序（row index = 1）
print('按第2行排序\n',arr[:,arr[1,:].argsort()])

arr
[[1 3 5]
 [4 2 0]]
按第2列排序
 [[4 2 0]
 [1 3 5]]
按第2行排序
 [[5 3 1]
 [0 2 4]]

 

附一个DataFrame排序的示意