如何在Matplotlib 3D Quiver图中高亮切片并提取z=0的2D切片？

核心思路 #

要获取z=0处的箭头切片，只需定位网格中z坐标接近0的层，提取该层对应的平面坐标（x、y）和箭头的水平分量（u、v），再用Matplotlib的2D版quiver函数绘制即可。

修改后的代码片段 #

替换原代码中从ax.quiver(...)到plt.show()的部分，改为以下内容：

# 定位z值接近0的层索引（避免数值精度问题，不用直接判断z==0）
z_slice_idx = np.where(np.isclose(z, 0, atol=1e-5))[2][0]

# 提取z=0层的坐标和箭头分量
x_slice = x[:, :, z_slice_idx]
y_slice = y[:, :, z_slice_idx]
u_slice = u[:, :, z_slice_idx]
v_slice = v[:, :, z_slice_idx]

# 创建2D绘图窗口
plt.figure()
ax_2d = plt.gca()
# 绘制2D箭头图，scale参数可调整箭头长度
ax_2d.quiver(x_slice, y_slice, u_slice, v_slice, scale=5, color='#1f77b4')

# 设置图表细节
ax_2d.set_xlabel('X')
ax_2d.set_ylabel('Y')
ax_2d.set_title('2D Quiver Slice at z=0')
# 保持坐标轴比例一致，防止箭头变形
ax_2d.set_aspect('equal')

plt.show()

关键细节说明 #

• 使用np.isclose匹配z值：由于数值计算可能存在微小误差，直接用z == 0容易漏匹配，加atol参数设置容差更可靠
• 切片维度对应：meshgrid生成的数组维度为(y, x, z)，所以第三个维度对应z层，提取时取[:, :, z_slice_idx]即可得到整个xy平面的数据
• 2D quiver参数：scale用于控制箭头长度，可根据可视化效果自行调整；设置aspect='equal'能保证箭头方向和比例的准确性

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