python - 字典排序浮点数组python

好的，我有一个4x2 NUMPY NDARRY，我想对它进行词典分类。也就是说，如果我有阵列

[[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]])