对包含numpy.ndarray元素的列表进行排序

时间:2019-05-05 03:01:45

标签: python arrays sorting opencv

我正在研究计算机视觉项目,我想问你一些问题。我使用cv2.findContours()方法,然后使用approxPolyDP(),每种形状都有4-4个检测到的边缘。图片上有三个彼此相邻的矩形。问题是,我想根据第一个x,y坐标对列表进行排序。从左到右。

谢谢!

contours, _ = cv2.findContours(raw_image2, cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)

contour_list = []
for contour in contours:
    approx= cv2.approxPolyDP(contour,0.1*cv2.arcLength(contour,True),True)
    contur_list.append(approx)


   [array([[[383,  22]],

   [[384, 127]],

   [[492, 127]],

   [[491,  20]]], dtype=int32), array([[[ 54,  16]],

   [[ 52, 123]],

   [[160, 124]],

   [[160,  17]]], dtype=int32), array([[[222,  14]],

   [[220, 124]],

   [[328, 125]],

   [[328,  15]]], dtype=int32)]

这是未排序的输出,但我希望成为这样:

   [array([[[ 54,  16]],

   [[ 52, 123]],

   [[160, 124]],

   [[160,  17]]],dtype=int32), array([[[222,  14]],

   [[220, 124]],

   [[328, 125]],

   [[328,  15]]], dtype=int32), array([[[383,  22]],

   [[384, 127]],

   [[492, 127]],

   [[491,  20]]], dtype=int32)]

1 个答案:

答案 0 :(得分:0)

像这样尝试内置函数sorted

sorted(contur_list, key=lambda approx: approx[0,0,0])

此处的key参数指定在进行比较之前在每个列表元素上要调用的函数。如果您想要反向排序列表的版本,可以调用reversed函数。