python - 查找3D表面的开口

我有一个包含笛卡尔坐标XYZ的数组。每一行是物体表面的一个点。我想找到那个表面的开口，旋转物体，使X轴指向开口。
我正在使用python和numpy，但是一般的方法和特定的实现一样好。
这是我目前的情况。X轴是红色的，原色是绿色的：

我想得到的是：

最佳答案

通常，您希望将旋转矩阵应用于数据。但是，还需要找到旋转矩阵。
在这种情况下，更容易直接跳到处理协方差矩阵的特征向量。这基本上是一种主成分方法。如果我们确定数据的主要组成部分并将其旋转到该坐标系中，我们将有效地完成您想要的操作。
首先，让我们生成一个与您类似的示例：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

def main():
    x, y, z = generate_data()
    plot(x, y, z)
    plt.show()

def generate_data():
    lat, lon = np.radians(np.mgrid[-90:90:20j, 0:180:20j])
    lon -= np.radians(40)
    z = np.cos(lat) * np.cos(lon)
    x = np.cos(lat) * np.sin(lon)
    y = np.sin(lat)
    return x, y, z

def plot(x, y, z):
    fig, ax = plt.subplots(subplot_kw=dict(projection='3d'), facecolor='w')
    artist = ax.scatter(x, y, z, marker='o', color='y')
    ax.set(xlim=[-1.1, 1.1], ylim=[-1.1, 1.1], zlim=[-1.1, 1.1], aspect=1)
    ax.set(xlabel='X', ylabel='Y', zlabel='Z')
    return artist

main()

现在我们可以根据主坐标旋转物体：

def reorient(x, y, z):
    xyz = np.vstack([x.ravel(), y.ravel(), z.ravel()])
    cov = np.cov(xyz)

    # Find the eigenvectors of the covariance matrix
    vals, vecs = np.linalg.eigh(cov)
    idx = np.argsort(vals)
    # The eigenvalues vals are not needed below, but this puts them in
    # the same order as the eigenvectors, should they be needed in future
    # versions of this code:
    vals, vecs = vals[idx], vecs[:, idx]

    # In this case, we actually want the second eigenvector to be the x-axis
    vecs = vecs[:, [1, 0, 2]]

    # Now let's perform a change-of-basis into the new coordinate system
    return np.linalg.inv(vecs).dot(xyz)

并绘制结果：

def main():
    x, y, z = generate_data()
    plot(*reorient(x, y, z))
    plt.show()

一个注意事项：我隐式地假设您的数据已经位于旋转发生的点的中心。如果不是这样的话，你需要在计算协方差矩阵之前减去旋转点（比如平均值），然后在改变基后再加上它。