如何用Mayavi在立方体各面绘制等高线图？技术实现咨询

我来帮你搞定用Mayavi实现立方体各面等高线的需求，还能解决你提到的交互问题，比Matplotlib的体验好太多啦～

一、核心思路：用contour_surf结合平面定位实现每个面的独立等高线 #

你之前尝试的contour_surf、surf其实完全可以实现需求，关键是要给每个立方体面单独生成对应平面的网格数据，把等高线“贴”在立方体的六个面上。Mayavi的3D渲染会自动处理遮挡关系，不会出现Matplotlib里前景面混淆的问题。

二、步骤1：生成立方体各面的网格与数据 #

首先我们为立方体的6个面分别创建对应的三维网格，每个面固定一个坐标轴的值（比如x=0、x=1），另外两个轴生成二维网格，再配上你的等高线数据：

import numpy as np
from mayavi import mlab

# 生成基础二维网格，对应立方体一个面的边长范围[0,1]
xx, yy = np.mgrid[0:1:10j, 0:1:10j]
# 替换成你自己的真实数据，这里用随机数示例
contour_data = np.random.randint(5, size=xx.shape)

# 定义立方体6个面的坐标映射：(x坐标, y坐标, z坐标, 等高线数据)
cube_faces = [
    # x=0面（左侧面）：y-z平面
    (np.zeros_like(xx), yy, xx, contour_data),
    # x=1面（右侧面）：y-z平面
    (np.ones_like(xx), yy, xx, contour_data),
    # y=0面（前侧面）：x-z平面
    (xx, np.zeros_like(yy), yy, contour_data),
    # y=1面（后侧面）：x-z平面
    (xx, np.ones_like(yy), yy, contour_data),
    # z=0面（底面）：x-y平面
    (xx, yy, np.zeros_like(xx), contour_data),
    # z=1面（顶面）：x-y平面
    (xx, yy, np.ones_like(xx), contour_data),
]

三、步骤2：绘制立方体各面的等高线 #

用mlab.contour_surf逐个绘制每个面的等高线，还可以指定等高线数量和颜色映射：

# 初始化Mayavi画布
mlab.figure(size=(800, 600), bgcolor=(1,1,1))

# 遍历每个面绘制等高线
for face in cube_faces:
    x, y, z, data = face
    # contours参数指定等高线数量，colormap选你喜欢的配色
    mlab.contour_surf(x, y, z, data, contours=5, colormap='coolwarm')

# 可选：添加立方体线框辅助观察，让结构更清晰
mlab.outline(extent=[0,1,0,1,0,1], color=(0,0,0), line_width=2)

# 调整视角到和你Matplotlib示例类似的角度
mlab.view(azimuth=-135, elevation=-22)

mlab.show()

运行这段代码就能得到和你预期效果一致的3D立方体等高线图，渲染和旋转交互都会非常流畅。

四、步骤3：实现点击获取坐标与等高线值的交互 #

Mayavi的交互API可以轻松实现点击获取坐标的功能，下面是一个简单的回调函数示例，点击任意面都会输出对应信息：

def on_face_click(picker):
    # 获取点击的对象
    clicked_obj = picker.actor
    if not clicked_obj:
        return
    
    # 获取点击的三维坐标
    x_3d, y_3d, z_3d = picker.point
    tol = 1e-3  # 坐标误差容忍度
    face_info = ""
    u, v = 0, 0
    data_value = 0

    # 判断点击的是哪个面，并映射到二维坐标
    if abs(x_3d - 0) < tol:
        face_info = "x=0左侧面"
        u, v = y_3d, z_3d
        data_value = contour_data[int(v*9), int(u*9)]  # 对应10j网格的索引范围0-9
    elif abs(x_3d - 1) < tol:
        face_info = "x=1右侧面"
        u, v = y_3d, z_3d
        data_value = contour_data[int(v*9), int(u*9)]
    elif abs(y_3d - 0) < tol:
        face_info = "y=0前侧面"
        u, v = x_3d, z_3d
        data_value = contour_data[int(v*9), int(u*9)]
    elif abs(y_3d - 1) < tol:
        face_info = "y=1后侧面"
        u, v = x_3d, z_3d
        data_value = contour_data[int(v*9), int(u*9)]
    elif abs(z_3d - 0) < tol:
        face_info = "z=0底面"
        u, v = x_3d, y_3d
        data_value = contour_data[int(v*9), int(u*9)]
    elif abs(z_3d - 1) < tol:
        face_info = "z=1顶面"
        u, v = x_3d, y_3d
        data_value = contour_data[int(v*9), int(u*9)]
    
    print(f"点击了{face_info}，二维坐标(u,v)=({u:.2f},{v:.2f})，等高线值={data_value}")

# 绑定点击事件
fig = mlab.gcf()
fig.scene.picker.pick_mode = 'point'
fig.scene.on_mouse_pick(on_face_click)

mlab.show()

五、关于拖放对象的扩展 #

如果要实现拖放对象到立方体面上的功能，可以结合Mayavi的on_mouse_down、on_mouse_move、on_mouse_up事件跟踪鼠标轨迹，在拖放结束时调用上面的坐标获取逻辑，再将对象放置到对应位置即可，Mayavi的交互API足够支持这个需求。

为什么选Mayavi？ #

Mayavi是Python专门用于3D科学可视化的库，基于VTK开发，渲染效果和交互性能远优于Matplotlib的3D模块，完全适配你需要精细3D交互的场景。如果追求极致定制化也可以直接用VTK，但Mayavi的API更简洁上手。

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