Question

假设我有一组积分，

R = [[x1, y1, z1],[x2, y2, z2],...,[xn, yn, zn]]

对于R中的每个点（p），我使用scipy.cKDTree确定了半径（r）和高度（2r）的局部邻域

import numpy as np
import scipy.spatial

R = np.array(R)
r = 1 # search radius

xy = R[:,0:2] # make array of ONLY xy
tree = scipy.spatial.cKDTree(xy)

for p in range(len(R)):
    2d_nbr_indices = tree.query_ball_point(xy[p], r) # indices w/in xy neighborhood
    2d_nbr_array = R[2d_nbr_indices] # 3d array of 2d neighbors
    z = R[p][1] # get z value
    zMin = z - r
    zMax = z + r
    # Create boolean array to filter 3d array
    hgt_filter = np.any([2d_nbr_array[:, 2] >= zMin,
                         2d_nbr_array[:, 2] <= zMax], axis=0)
    3d_nbr_array = 2d_nbr_array[hgt_filter] # points in xyz neighborhood

我想计算每个邻域的正交回归平面，确定从每个点到平面的距离（正交），并计算平面的法向量。有没有人对如何在python中进行此操作有任何建议？

编辑：我找到了odr user guide。它似乎处理3d点。欢迎任何关于它的实施和使用的建议。我还找到了this similar question。

编辑：我应该提到数据可能包含垂直或近乎垂直的表面，因此需要隐式模型。我找到了this example in the scipy codebook，但只发现了xy数据。

Answer 1

这是如何使用scipy.odr将三维表面与点云拟合的一般示例。希望它有所帮助。

from scipy.odr import ODR, Model, Data
import numpy as np
import matplotlib.pyplot as plt

from mpl_toolkits.mplot3d import Axes3D

def func(beta,data):
    x,y = data
    a,b,c = beta
    return a*x+b*y+c

N = 20
x = np.random.randn(N)
y = np.random.randn(N)
z = func([-3,-1,2],[x,y])+np.random.normal(size=N)


data = Data([x,y],z)
model = Model(func)
odr = ODR(data, model, [0.0,0.0,0.0])
odr.set_job(fit_type = 0)
res = odr.run()

Y,X = np.mgrid[y.min():y.max():20j,x.min():x.max():20j]
Z = func(res.beta, [X,Y])
f = plt.figure()
pl = f.add_subplot(111,projection='3d')
pl.scatter3D(x,y,z)
pl.plot_surface(X,Y,Z,alpha=0.4)

如何估算3d点的局部切平面？

1 个答案: