python - 在python中按行对二维数组进行排序

Question

我想根据第一列的降序重新排列 python numpy 数组中的行。例如，

([[2,3,1,8],
  [4,7,5,20],
  [0,-2,2,0]])

转换为

([[0,-2,2,0],
  [2,3,1,8],
  [4,7,5,20]])

第一列转换为 [0,2,4]来自 [2,4,0]

Answer 1

常规 sorted 是这样做的:

print(sorted([[2,3,1,8], [4,7,5,20], [0,-2,2,0]]))

但如果您只想按第一列排序，请使用:

print(sorted([[2,3,1,8], [4,7,5,20], [0,-2,2,0]], key=lambda x: x[0]))

它们都输出:

[[0, -2, 2, 0], [2, 3, 1, 8], [4, 7, 5, 20]]