在numpy中找到两个多边形之间的距离

时间:2019-10-18 10:10:03

标签: python numpy geometry

我有两个多边形P和Q,其中多边形的外部线性环由两个封闭的点集(存储为numpy数组)定义,它们以逆时针方向连接。 P和Q的格式如下:

P['x_coords'] = [299398.56 299402.16 299410.25 299419.7  299434.97 299443.75 299454.1 299465.3  299477.   299488.25 299496.8  299499.5  299501.28 299504. 299511.62 299520.62 299527.8  299530.06 299530.06 299525.12 299520.2 299513.88 299508.5  299500.84 299487.34 299474.78 299458.6  299444.66 299429.8  299415.4  299404.84 299399.47 299398.56 299398.56] 
P['y_coords'] = [822975.2  822989.56 823001.25 823005.3  823006.7  823005.06 823001.06 822993.4  822977.2  822961.   822943.94 822933.6  822925.06 822919.7 822916.94 822912.94 822906.6  822897.6  822886.8  822869.75 822860.75 822855.8  822855.4  822857.2  822863.44 822866.6  822870.6  822876.94 822886.8  822903.   822920.3  822937.44 822954.94 822975.2]

Q['x_coords'] = [292316.94 292317.94 292319.44 292322.47 292327.47 292337.72 292345.75 292350.   292352.75 292353.5  292352.25 292348.75 292345.75 292342.5 292338.97 292335.97 292333.22 292331.22 292329.72 292324.72 292319.44 292317.2  292316.2  292316.94]
Q['y_coords'] = [663781.   663788.25 663794.   663798.06 663800.06 663799.3  663796.56 663792.75 663788.5  663782.   663773.25 663766.   663762.   663758.25 663756.5  663756.25 663757.5  663761.   663763.75 663767.5  663769.5 663772.25 663777.5  663781.  ]

## SIMPLIFIED AND FORMATTED FOR EASY TESTING:
import numpy as np

px_coords = np.array([299398,299402,299410.25,299419.7,299398])
py_coords = np.array([822975.2,822920.3,822937.44,822954.94,822975.2])

qx_coords = np.array([292316,292331.22,292329.72,292324.72,292319.44,292317.2,292316])
qy_coords = np.array([663781,663788.25,663794,663798.06,663800.06,663799.3,663781])

P的外环是通过连接P['x_coords'][0], P['y_coords'][0] -> P['x_coords'][1], P['y_coords'][1]等形成的。每个数组的最后一个坐标与第一个数组相同,表示形状在拓扑上是封闭的。

是否可以使用numpy计算P和Q的外圈之间的简单最小距离?我一直在搜索SO上下两步,而没有找到任何明确的内容,因此我怀疑这可能是一个非常复杂的问题的过度简化。我知道可以使用GDAL或Shapely等现成的空间库来进行距离计算,但是我很想通过在numpy中从头开始构建一些东西来了解它们的工作原理。

我已经考虑或尝试过的一些事情:

  • 计算两个数组中每个点之间的距离。这不起作用,因为P和Q之间的最接近点可以是边顶点对。使用scipy.spatial计算的每种形状的凸包都存在相同的问题。
  • 一种低效的蛮力方法,计算每对点之间以及每个边点对组合之间的距离

是否有更好的方法来解决此问题?

1 个答案:

答案 0 :(得分:1)

k - d 树上有many variations,用于存储具有范围的对象,例如 edges >个多边形。我最熟悉(但没有链接)的方法涉及将轴对齐的边界框与每个节点相关联。叶子与对象相对应,内部节点的框是包围两个子节点的框(通常重叠)的最小框。通常的中位数切割方法适用于对象框的中点(对于线段,这是它们的中点)。

为每个多边形构建了这些方法之后,以下双重递归查找最接近的方法:

def closest(k1,k2,true_dist):
  return _closest(k1,0,k2,0,true_dist,float("inf"))

def _closest(k1,i1,k2,i2,true_dist,lim):
  b1=k1.bbox[i1]
  b2=k2.bbox[i2]
  # Call leaves their own single children:
  cc1=k1.child[i1] or (i1,)
  cc2=k2.child[i2] or (i2,)
  if len(cc1)==1 and len(cc2)==1:
    return min(lim,true_dist(i1,i2))
  # Consider 2 or 4 pairs of children, possibly-closest first:
  for md,c1,c2 in sorted((min_dist(k1.bbox[c1],k2.bbox[c2]),c1,c2)
                         for c1 in cc1 for c2 in cc2):
    if md>=lim: break
    lim=min(lim,_closest(k1,c1,k2,c2,true_dist,lim)
  return lim

注意:

  • 两个不相交的线段之间最接近的方法true_dist必须涉及至少一个端点。
  • 点与线段之间的距离可以大于点与包含线段的线之间的距离。
  • 不需要点对点检查:可以通过相邻的边缘找到这样的一对(四次)。
  • min_dist的边界框参数可能是重叠的,在这种情况下,它必须返回0。
相关问题