好的,所以我有一个 4x2 numpy ndarray,我想按字典顺序对其进行排序。也就是说,如果我有数组
[[0,0],
[1,1],
[0,1],
[1,0]]
我希望它成为
[[0,0],
[0,1],
[1,0],
[1,1]]
我该怎么做?
您可以使用
numpy的lexsort
。 但是,Lexsort 使用最后一列作为主键进行排序。 获取所需内容的一种方法是显式指定列:
x[np.lexsort((x[:,1], x[:,0]))]
# array([[0, 0],
# [0, 1],
# [1, 0],
# [1, 1]])