Python：定义最小边界矩形

Question

我有以下格式的数据，二维 x,y 坐标列表：

[(6, 7), (2, 4), (8, 9), (3, 7), (5, 4), (9, 9)]

我正在尝试遍历列表以找到格式为 [(minx,miny),(maxx,miny),(maxx,maxy),(minx,maxy)] 的最小边界框

因此，我编写了以下代码，但它似乎不起作用。可能与我没有将其传递给数组而不是列表的事实有关。输入和代码如下，打印坐标返回前面提到的列表：

os.chdir("filepath")
with open ('file.csv') as csvfile:
    coords = [(int(x), int(y)) for x, y in csv.reader(csvfile, delimiter= ',')]
print coords


def bounding_box(coords):
    min_x, min_y = numpy.min(coords[0], axis=0)
    max_x, max_y = numpy.max(coords[0], axis=0)
    return numpy.array(
        [(min_x, min_y), (max_x, min_y), (max_x, max_y), (min_x, max_y)])

谁能指出我正确的方向？如果没有帮助，请随意忽略上面的整个代码。

Answer 1

为什么不用四个计数器遍历列表：min_x、min_y、max_x 和 max_y

def bounding_box(coords):

  min_x = 100000 # start with something much higher than expected min
  min_y = 100000
  max_x = -100000 # start with something much lower than expected max
  max_y = -100000

  for item in coords:
    if item[0] < min_x:
      min_x = item[0]

    if item[0] > max_x:
      max_x = item[0]

    if item[1] < min_y:
      min_y = item[1]

    if item[1] > max_y:
      max_y = item[1]

  return [(min_x,min_y),(max_x,min_y),(max_x,max_y),(min_x,max_y)]

使用您的示例输入：

bounding_box([(6, 7), (2, 4), (8, 9), (3, 7), (5, 4), (9, 9)])
>> [(2, 4), (9, 4), (9, 9), (2, 9)]

Answer 2

您遇到的问题是您正在搜索矢量最小值/最大值，而您应该按每个 verctor 维度（在您的情况下为两个）进行搜索。 所以你应该寻找 X 的最小值和 Y 的最小值，最大值相同。

正确的做法是：

min_x, min_y = numpy.min(coords[...,0]), numpy.min(coords[...,1])
max_x, max_y = numpy.max(coords[...,0]), numpy.max(coords[...,1])
return numpy.array(
    [(min_x, min_y), (max_x, min_y), (max_x, max_y), (min_x, max_y)])

coords[...,0] 和 coords[...,1] 表示您正在从特定维度（分别为 X 或 Y）中选择值。

为了让所有这些都能正常工作，您还应该将坐标列表设为一个 numpy 二维数组，而不是元组列表。