如何从左上角到右下角排列坐标？

Question

我有一个类对象 my_rectangle 的列表：

class my_rectangle:
    def __init__(self,text,x_start,y_start,x_end,y_end):
        self.text=text
        self.x_start=x_start
        self.y_start=y_start
        self.x_end=x_end
        self.y_end=y_end
        self.x_centroid=(self.x_start+self.x_end)/2
        self.y_centroid=(self.y_start+self.y_end)/2

使用给出质心坐标的类属性（x_centroid 和y_centroid），我想排序这个列表，使用从左到右然后从上到下的顺序（普通英文阅读顺序）？

说我有：

A=my_rectangle('Hi,',1,3,2,4)
B=my_rectangle('Im',3,3,3,4)
C=my_rectangle('New',1,1,2,2)
my_list=[C,B,A]

我想订购它来获得：

my_sorted_list=[A,B,C]

这是文本的表示：

""" Hi, I'm
    New 
"""

Answer 1

生成排序列表是内置函数sorted()的专长。

可以通过提供key 函数来完成使用多个值的排序，该键函数将值作为元组返回。然后根据元组的字典顺序对结果列表进行排序。

#UNTESTED
my_sorted_list = sorted(my_list, key=lambda item: (item.x_centroid, item.y_centroid))

Answer 2

您可以通过定义__lt__ 方法使自定义类可排序。这会处理默认排序中使用的< 运算符。

class Rectangle:
    def __init__(self,text,x_start,y_start,x_end,y_end):
        self.text=text
        self.x_start=x_start
        self.y_start=y_start
        self.x_end=x_end
        self.y_end=y_end

    @property
    def centroid(self):
        return (self.x_start+self.x_end)/2, (self.y_start+self.y_end)/2

    def __lt__(self, other):
        """Using "reading order" in a coordinate system where 0,0 is bottom left"""
        try:
            x0, y0 = self.centroid
            x1, y1 = other.centroid
            return (-y0, x0) < (-y1, x1)
        except AttributeError:
            return NotImplemented

    def __repr__(self):
        return 'Rectangle: ' + self.text

我将centroid 定义为一个属性，以便在初始化矩形后更改任何其他坐标时它会更新。

如果您使用问题中的数据，您将获得此输出。

>>> rectangles = [
...     Rectangle('A',1,3,2,4),
...     Rectangle('B',3,3,3,4),
...     Rectangle('C',1,1,2,2), 
... ]
>>> print(sorted(rectangles))
[Rectangle: A, Rectangle: B, Rectangle: C]