python - 用numpy插值一个角度的旋转

我只有一点

x = ..
y = ..
p = np.matrix([[x],[y]])

我希望以 d 度为增量围绕原点旋转该点，以获得 N = 360/d 点。例如，我想象以下功能。

points = interpolate360(d, p)

点的形状应为(2，N)

我可以使用一个新的旋转矩阵为每个旋转循环执行代码，然后合并结果，但是我希望有某种矢量化的解决方案。

最佳答案

在大多数设置中，使用numpy的matrix可能不是最好的主意。解决问题的一种方法是创建一个3D数组，其中[n, :, :]保存第n个角度的旋转矩阵。您无法使用3D matrix，因此，如果您混合使用数组和矩阵类型，并且仍然想依靠*进行矩阵乘法，则可能会变得困惑。如果您坚持使用数组，并使用np.dot来可预测地处理矩阵乘法，则以下代码可以很好地工作。实际上，它也需要一个matrix，但首先将其转换为ndarray:

def interpolate360(d, p):
    p = np.array(p)
    angles = np.arange(0, 2 * np.pi, d * np.pi / 180)
    sin = np.sin(angles)
    cos = np.cos(angles)

    rot_matrices = np.empty((angles.shape[0], 2, 2))
    rot_matrices[..., 0, 0] = cos
    rot_matrices[..., 0, 1] = -sin
    rot_matrices[..., 1, 0] = sin
    rot_matrices[..., 1, 1] = cos

    return np.dot(rot_matrices, p)

如下例所示，如果您的输入是一维行向量，二维单列向量或包含多个列向量的二维数组，则此方法有效:

>>> interpolate360(90, [0, 1])
array([[  0.00000000e+00,   1.00000000e+00],
       [ -1.00000000e+00,   6.12323400e-17],
       [ -1.22464680e-16,  -1.00000000e+00],
       [  1.00000000e+00,  -1.83697020e-16]])
>>> interpolate360(90, [[0], [1]])
array([[[  0.00000000e+00],
        [  1.00000000e+00]],

       [[ -1.00000000e+00],
        [  6.12323400e-17]],

       [[ -1.22464680e-16],
        [ -1.00000000e+00]],

       [[  1.00000000e+00],
        [ -1.83697020e-16]]])
>>> interpolate360(90, [[1, 0], [0, 1]])
array([[[  1.00000000e+00,   0.00000000e+00],
        [  0.00000000e+00,   1.00000000e+00]],

       [[  6.12323400e-17,  -1.00000000e+00],
        [  1.00000000e+00,   6.12323400e-17]],

       [[ -1.00000000e+00,  -1.22464680e-16],
        [  1.22464680e-16,  -1.00000000e+00]],

       [[ -1.83697020e-16,   1.00000000e+00],
        [ -1.00000000e+00,  -1.83697020e-16]]])

关于python - 用numpy插值一个角度的旋转，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/14232588/