numpy根据两个向量计算旋转矩阵

在NumPy中，计算根据两个向量旋转矩阵通常涉及到使用二维欧几里得空间中的旋转变换。假设我们有两个单位正交向量`u`和`v`，我们可以按照Rodrigues公式来构造旋转矩阵。这个公式基于三个方向向量，但如果已知两个，可以先通过它们构成新的坐标系，然后找到第三个。

首先，我们需要确定这两个向量的夹角`theta`，通常是通过向量点积和余弦函数来计算：

import numpy as np

def angle_between(v1, v2):
    dot_product = np.dot(v1, v2)
    norm_v1 = np.linalg.norm(v1)
    norm_v2 = np.linalg.norm(v2)
    return np.arccos(dot_product / (norm_v1 * norm_v2))

u, v = ... # 输入的两个向量
theta = angle_between(u, v)

接下来，我们可以构造一个临时矩阵`R`，它代表绕`u`轴旋转90度的变换，再根据`theta`进行缩放和平移：

R_axis = np.array([[0, -u[2], u[1]], [u[2], 0, -u[0]], [-u[1], u[0], 0]])
R = np.eye(3) + np.sin(theta)*R_axis + (1 - np.cos(theta))*(np.dot(R_axis, R_axis))

这里的`eye(3)`是一个3x3的单位矩阵，`np.sin(theta)*R_axis`和`(1 - np.cos(theta))*(np.dot(R_axis, R_axis))`分别对应于旋转和缩放的部分。

注意，这得到的是绕`u`轴的旋转矩阵，如果你想得到绕`v`轴的旋转矩阵，你需要先应用上述操作到`u`上，然后再旋转一次。