首页 > 解决方案 > 如何有效地从python中的3个变形点找到旋转矩阵

问题描述

我试图通过使用我在两个坐标系中定义的 3 个单位向量来找到坐标系之间的旋转矩阵。 坐标旋转可视化

我目前正在使用我在 Mathmatics 上找到的方法,其定义如下: 旋转矩阵方程

在 python 中实现它的最小工作示例如下所示:

p1 = [math.sqrt(2)/2,math.sqrt(2)/2,0]
p2 = [-math.sqrt(2)/2,math.sqrt(2)/2,0]
p3 = [0,0,1]
TT = np.vstack([p1, p2, p3]).T
TM = np.eye(3)
for i in range (0,2):
    for j in range(0,2):
        TM[i][j] = math.cos(angle_between_vectors(np.eye(3)[:,j],TT[i,:]))


def vector_norm(data, axis=None, out=None):
    data = np.array(data, dtype=np.float64, copy=True)
    if out is None:
        if data.ndim == 1:
            return math.sqrt(np.dot(data, data))
        data *= data
        out = np.atleast_1d(np.sum(data, axis=axis))
        np.sqrt(out, out)
        return out
    else:
        data *= data
        np.sum(data, axis=axis, out=out)
        np.sqrt(out, out)

def angle_between_vectors(v0, v1, directed=True, axis=0):
    v0 = np.array(v0, dtype=np.float64, copy=False)
    v1 = np.array(v1, dtype=np.float64, copy=False)
    dot = np.sum(v0 * v1, axis=axis)
    dot /= np.linalg.norm(v0, axis=axis) * vector_norm(v1, axis=axis)
    dot = np.clip(dot, -1.0, 1.0)
    return np.arccos(dot if directed else np.fabs(dot)) 

有谁知道用更少的代码或更高的效率计算旋转矩阵的方法?

标签: pythonnumpymathlinear-algebra

解决方案


推荐阅读