返回numpy meshgrid中每个网格方块的边界框

时间:2017-09-02 18:48:42

标签: python numpy coordinates

我有一个带边界框(左下和右上坐标)的函数,以及在每个轴上将网格分割成网格方格的数量。该函数返回表示每个网格方块左下角的坐标列表。

def grid(min_x, min_y, max_x, max_y, split_grid_x, split_grid_y):
    x = np.linspace(min_x, max_x, split_grid_x)
    y = np.linspace(min_y, max_y, split_grid_y)
    xv, yv = np.meshgrid(x, y)
    grid = np.hstack((xv.reshape((xv.size, 1)), yv.reshape((yv.size, 1))))
    return grid.tolist()

例如:

grid(0, 0, 2, 2, 3, 3)

返回:

[[0.0, 0.0], [1.0, 0.0], [2.0, 0.0], [0.0, 1.0], [1.0, 1.0], [2.0, 1.0], [0.0, 2.0], [1.0, 2.0], [2.0, 2.0]]

我想要的是函数返回每个网格方块的边界框。例如,上面数组中的第一个元素将返回[0.0,0.0,1.0,1.0]

我希望我已经有效地解释了这一点,如果您需要澄清,请告诉我。提前谢谢!

0 个答案:

没有答案