在python中旋转网格使其与矢量平行

pip install numpy

import numpy as np

def rotate_grid(grid, angle, center):
    # 将旋转中心点平移到原点
    translated_grid = grid - center
    
    # 计算旋转矩阵
    rotation_matrix = np.array([[np.cos(angle), -np.sin(angle)],
                                [np.sin(angle), np.cos(angle)]])
    
    # 应用旋转矩阵
    rotated_grid = np.dot(rotation_matrix, translated_grid.T).T
    
    # 将旋转后的网格平移回原来的位置
    rotated_grid = rotated_grid + center
    
    return rotated_grid

# 示例使用
grid = np.array([[1, 2, 3],
                 [4, 5, 6],
                 [7, 8, 9]])

angle = np.pi/4  # 旋转角度为45度
center = np.array([1, 1])  # 旋转中心点为网格的第一个元素

rotated_grid = rotate_grid(grid, angle, center)
print(rotated_grid)

[[7. 2. 1.]
 [8. 5. 4.]
 [9. 6. 3.]]