Mayavi合并3D等高线图与箭头图时坐标不匹配问题求助

问题根源 #

• 坐标索引不一致：等高线图使用indexing='ij'生成网格坐标，箭头的网格坐标却用了默认的indexing='xy'，导致两者的坐标维度映射混乱，Mayavi无法统一坐标系。
• 不必要的meshgrid使用：单个箭头无需用meshgrid处理坐标，生成的(1,1,1)形状数组会干扰场景缩放计算，让箭头显示极小。
• 箭头缩放未适配场景：默认缩放规则在0-5的大坐标范围下，会导致箭头被过度缩小。

修复步骤 #

1. 统一坐标逻辑：等高线图直接传入原始1D坐标数组，Mayavi的contour3d会自动构建对应网格，无需手动生成xx_arr,yy_arr,zz_arr。
2. 简化箭头输入：单个箭头直接用1D数组作为起点和向量参数，无需meshgrid处理。
3. 调整箭头缩放：通过scale_factor设置箭头缩放比例，搭配scale_mode='none'避免自动缩放干扰，确保箭头在场景中清晰可见。
4. 强制场景范围识别：添加轮廓线让Mayavi正确识别整个场景的坐标区间，避免坐标轴显示异常。

修改后的代码 #

import numpy as np
from mayavi import mlab

# 箭头起点坐标
x = np.array([2.])
y = np.array([1.])
z = np.array([0.])

# 箭头向量分量
u = np.array([3.])
v = np.array([0.])
w = np.array([0.])

# 球体坐标网格
x_arr = np.linspace(0,5,50)
y_arr = np.linspace(0,5,50)
z_arr = np.linspace(0,5,50)

# 球体中心和半径
xc = 2
yc = 2.5
zc = 3
r  = 2

# 生成球体数据
data2plot = np.zeros((len(x_arr), len(y_arr), len(z_arr)))
for ii in range(len(x_arr)):
    for jj in range(len(y_arr)):
        for kk in range(len(z_arr)):
            if np.sqrt((x_arr[ii]-xc)**2 + (y_arr[jj]-yc)**2 + (z_arr[kk]-zc)**2) < r:
                data2plot[ii,jj,kk] = 1

fig1 = mlab.figure(bgcolor=(1,1,1), fgcolor=(0,0,0))

# 绘制等高线图：直接传入1D坐标数组
contPlot = mlab.contour3d(x_arr, y_arr, z_arr, data2plot, 
                          transparent=True, opacity=.4, figure=fig1)

# 绘制箭头：设置缩放参数确保显示正常
arrowPlot = mlab.quiver3d(x, y, z, u, v, w, 
                          line_width=1, mode='arrow', figure=fig1,
                          scale_factor=0.5, scale_mode='none')

# 添加坐标轴和轮廓线，修正场景范围
ax1 = mlab.axes(color=(0,0,0), nb_labels=4)
mlab.outline(color=(0,0,0))

mlab.show()

额外说明 #

• 若需绘制多个箭头，只需将x,y,z,u,v,w扩展为包含多个点的1D数组即可，无需meshgrid。
• scale_factor的取值可根据场景范围灵活调整，确保箭头大小适配。

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