Python中在热力图上绘制arctan2角度箭头的问题

核心问题拆解 #

• quiver 不是直接接收角度值，它需要的是坐标起点 + x/y方向的向量分量。你直接传arctan2的结果当向量参数，方向肯定不对——其实diff_X和diff_Y本身就是对应角度的向量分量，直接用它们就行。
• 要让箭头大小大致恒定，得先把所有向量归一化成单位向量，再乘以固定的缩放系数，这样不管原向量模长多大，箭头尺寸都统一。

修正后的代码示例 #

import matplotlib.pyplot as plt
import numpy as np

# 假设你已经有diff_X、diff_Y这两个矩阵
diff = np.sqrt(diff_X**2 + diff_Y**2)

# 1. 绘制热力图，注意对齐坐标原点
plt.imshow(diff, cmap='viridis', origin='lower')
plt.colorbar(label='Magnitude')

# 2. 处理箭头：归一化向量，保证大小一致
magnitude = np.sqrt(diff_X**2 + diff_Y**2)
# 替换零值，避免除以零报错
magnitude[magnitude == 0] = 1e-8
# 归一化后乘以固定系数（0.5可按需调整）
U = diff_X / magnitude * 0.5
V = diff_Y / magnitude * 0.5

# 生成对应网格坐标，indexing='ij'确保和矩阵索引匹配
y, x = np.meshgrid(np.arange(diff.shape[0]), np.arange(diff.shape[1]), indexing='ij')

# 3. 绘制箭头
plt.quiver(x, y, U, V, color='white', scale=20, scale_units='xy', width=0.005)

plt.xlabel('X')
plt.ylabel('Y')
plt.title('Heatmap with Uniform Direction Arrows')
plt.show()

关键参数说明 #

• origin='lower'：imshow默认左上角是(0,0)，和quiver的坐标逻辑错位，加这个参数让矩阵的(0,0)对应图的左下角，坐标完全对齐。
• 向量归一化：diff_X/magnitude和diff_Y/magnitude得到单位向量，再乘固定系数，确保所有箭头大小一致。
• scale与scale_units='xy'：scale_units='xy'让箭头长度基于坐标轴单位，scale值越大箭头越短，可根据图的尺寸调整。
• width：控制箭头杆的粗细，避免遮挡热力图细节。

避坑提醒 #

• 别直接把arctan2的角度传给quiver，真要从角度转向量，得用U = np.cos(theta)、V = np.sin(theta)（theta是弧度值）。
• 网格坐标必须加indexing='ij'，不然箭头位置会和热力图的矩阵点错位。

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