仅当第一列中的值相同时，才根据第二列对numpy数组进行排序 [英] Sort a numpy array according to 2nd column only if values in 1st column are same

问题描述

我有一个numpy数组，其中包含使用OpenCV的findNonZero()方法获得的坐标.我想对它们进行排序以绘制轮廓.

I have a numpy array which contains coordinates got by using findNonZero() method of OpenCV. I want to sort them in order to draw contours.

未排序的numpy数组示例(非坐标):

 [[ 0, 2],
  [ 0, 0],
  [-1, 8],
  [-6, 7],
  [-1, 1]]

预期的排序numpy数组:

 [[-6, 7],
  [-1, 1],
  [-1, 8],
  [ 0, 0],
  [ 0, 2]]

我希望根据第一列对数组进行排序，如果第一列中的值相等，那么我希望根据第二列对数组进行排序.因此，由于处理是在云上完成的，所以时间复杂度最低吗?

I want the array to be sorted according to the first column and if values in the first column are equal, i want it to be sorted according to the second column. Is there a way to do so with least time complexity since the processing is to be done on cloud?

推荐答案

您可以使用 np.lexsort -

You can use np.lexsort -

a[np.lexsort(a[:,::-1].T)]

样品运行-

In [42]: a
Out[42]: 
array([[ 0,  2],
       [ 0,  0],
       [-1,  8],
       [-6,  7],
       [-1,  1]])

In [43]: a[np.lexsort(a[:,::-1].T)]
Out[43]: 
array([[-6,  7],
       [-1,  1],
       [-1,  8],
       [ 0,  0],
       [ 0,  2]])

这篇关于仅当第一列中的值相同时，才根据第二列对numpy数组进行排序的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！