如何旋转Meshgrid网格点并保持正交性（NumPy实现）

你遇到的旋转后网格视觉上不垂直的效果如下图：

这本质是操作和配置问题导致的假象。2D标准旋转矩阵是严格正交矩阵，属于刚体变换，变换过程不会改变任意两点间距、任意两条线段的夹角，原本正交的网格只要旋转计算逻辑正确，网格线必然保持90度垂直。出现"不垂直"的观感基本是踩了两个常见坑：

• 绘图时未开启坐标轴等比例：matplotlib默认会自动拉伸坐标轴适配画布尺寸，x轴和y轴上1单位长度对应的像素长度不一致，哪怕网格本身是正交的，视觉上也会出现夹角变形
• 点变换逻辑冗余易错：原代码把二维网格点展平成一维数组后用einsum做变换，维度对应关系绕弯，后续reshape回网格结构、绘制网格线时很容易搞反行列索引，连错点自然看起来网格变形

直接在meshgrid生成的二维网格结构上做变换，不要提前展平点，最后绘图时强制坐标轴等比例即可，参考实现如下：

import numpy as np
import matplotlib.pyplot as plt

# 基础参数
rot_degrees = 10
x = np.linspace(200, 600, 4)
y = np.linspace(200, 700, 5)
# 生成网格，明确指定xy索引模式避免行列歧义
xv, yv = np.meshgrid(x, y, indexing='xy')

# 若需要绕网格中心旋转而非绕坐标原点(0,0)旋转，先做坐标平移
center_x, center_y = x.mean(), y.mean()
pts_shifted = np.stack([xv - center_x, yv - center_y], axis=-1)  # 形状为(ny, nx, 2)，最后一维存坐标

# 构造旋转矩阵
radians = np.radians(rot_degrees)
rot_mat = np.array([
    [np.cos(radians), -np.sin(radians)],
    [np.sin(radians),  np.cos(radians)]
])

# 批量旋转所有点：直接用矩阵乘，写法直观不容易出错
pts_rotated_shifted = pts_shifted @ rot_mat.T

# 平移回原坐标位置
xv_rot = pts_rotated_shifted[..., 0] + center_x
yv_rot = pts_rotated_shifted[..., 1] + center_y

# 可视化验证
plt.figure(figsize=(6,6))
# 绘制旋转后的网格线：按行连是横向网格，按列连是纵向网格
plt.plot(xv_rot, yv_rot, c='blue', linewidth=1.5)
plt.plot(xv_rot.T, yv_rot.T, c='blue', linewidth=1.5)
# 绘制原始网格做对比
plt.plot(xv, yv, c='red', alpha=0.3, linestyle='--')
plt.plot(xv.T, yv.T, c='red', alpha=0.3, linestyle='--')
# 关键配置：强制坐标轴等比例，保证视觉上角度不畸变
plt.axis('equal')
plt.show()

• 不要提前把meshgrid生成的二维网格展平成一维点数组处理，直接在(ny, nx, 2)形状的数组上做变换，能从根源上避免索引错位、连线错误的问题
• 只要是正交旋转矩阵做的变换，就不会破坏网格的正交属性，不需要额外加正交修正逻辑
• 如果需要绕任意指定点旋转，只需要修改平移步骤的中心坐标即可，旋转后网格依然保持正交
• 只要是绘制涉及角度、比例的2D几何图形，都要加plt.axis('equal')，否则所有角度、长度的视觉表现都是错的

内容的提问来源于stack exchange，提问作者Jakub