python - 对二维数组进行排序时出现问题

Question

我有一个以下 2D numpy 数组:

array([[1 0]
       [2 0]
       [4 0]
       [1 1]
       [2 1]
       [3 1]
       [4 2])

我想用第二个值中的值对第一列的 ID 进行排序，我得到的结果很糟糕:

array([[1 0]
       [1 1]
       [2 0]
       [2 1]
       [3 1]
       [4 0]
       [4 2]])

我的复杂度为 O(n^2)，并希望进一步改进。

Answer 1

对列表列表进行排序的更好方法:

import numpy as np
a = np.array([[1, 0], [2 ,0], [4 ,0], [1 ,1], [2 ,1], [3 ,1], [4 ,2]])
s_a = np.asarray(sorted(a, key=lambda x: x[0]))
print(s_a)

输出:

[[1 0]
 [1 1]
 [2 0]
 [2 1]
 [3 1]
 [4 0]
 [4 2]]