如何实现Mayavi多场景切换与场景内绘图管理？

我来帮你搞定这个Mayavi多场景切换的需求！其实核心就是先把所有场景的绘图对象都创建好，然后通过按钮控制它们的显示/隐藏状态，确保同一时间只展示一个场景。下面是具体的实现方案和代码示例：

• 预创建所有场景的绘图对象：先把三个点云图都生成出来，初始设置为隐藏状态，避免切换时重复生成数据，提升流畅度
• 按钮触发切换逻辑：给每个按钮绑定一个回调函数，点击时将对应场景设为可见，同时隐藏其他所有场景
• 利用Mayavi对象的visible属性：所有Mayavi绘图对象（比如points3d返回的Actor）都自带这个属性，用来控制显示/隐藏，非常方便

import numpy as np
from mayavi import mlab

# ----------------------
# 第一步：生成测试数据（替换成你的实际数据）
# ----------------------
np.random.seed(42)
n_points = 100

# 场景1数据（蓝色）
x1 = np.random.rand(n_points)
y1 = np.random.rand(n_points)
z1 = np.random.rand(n_points)
s1 = np.random.rand(n_points)*10

# 场景2数据（红色）
x2 = np.random.rand(n_points) + 1
y2 = np.random.rand(n_points) + 1
z2 = np.random.rand(n_points) + 1
s2 = np.random.rand(n_points)*10

# 场景3数据（绿色）
x3 = np.random.rand(n_points) + 2
y3 = np.random.rand(n_points) + 2
z3 = np.random.rand(n_points) + 2
s3 = np.random.rand(n_points)*10

# ----------------------
# 第二步：创建所有绘图对象并初始隐藏
# ----------------------
fig = mlab.figure(title="多场景切换演示")

# 创建三个点云对象，初始都隐藏
cloud1 = mlab.points3d(x1, y1, z1, s1, color=(0, 0, 1), name="场景1")
cloud1.visible = False

cloud2 = mlab.points3d(x2, y2, z2, s2, color=(1, 0, 0), name="场景2")
cloud2.visible = False

cloud3 = mlab.points3d(x3, y3, z3, s3, color=(0, 1, 0), name="场景3")
cloud3.visible = False

# ----------------------
# 第三步：定义场景切换函数
# ----------------------
def show_scene1():
    # 显示场景1，隐藏其他
    cloud1.visible = True
    cloud2.visible = False
    cloud3.visible = False

def show_scene2():
    # 显示场景2，隐藏其他
    cloud1.visible = False
    cloud2.visible = True
    cloud3.visible = False

def show_scene3():
    # 显示场景3，隐藏其他
    cloud1.visible = False
    cloud2.visible = False
    cloud3.visible = True

# ----------------------
# 第四步：添加切换按钮到界面
# ----------------------
# 在Mayavi窗口添加三个按钮，分别绑定切换函数
mlab.add_button(show_scene1, label="按钮1", width=100, height=30)
mlab.add_button(show_scene2, label="按钮2", width=100, height=30)
mlab.add_button(show_scene3, label="按钮3", width=100, height=30)

# 默认显示第一个场景
show_scene1()

# 启动Mayavi界面
mlab.show()

• 预创建对象的优势：如果每次点击按钮才生成点云，数据量大的时候会有明显卡顿；预创建后只控制显示状态，切换几乎是即时的。
• 复杂场景的分组管理：如果每个场景包含多个绘图元素（比如点云+坐标轴+曲面），可以用mlab.pipeline.group()把同一场景的元素归为一组，然后控制组的visible属性，这样切换逻辑更简洁：

  # 示例：给场景1创建分组
  group1 = mlab.pipeline.group(name="场景1组")
  # 将该场景的所有元素都指定parent为group1
  cloud1 = mlab.points3d(x1, y1, z1, s1, color=(0,0,1), parent=group1)
  # 比如再加一个坐标轴
  mlab.axes(parent=group1)
  # 控制整个组的显示/隐藏
  group1.visible = False
  

• 按钮样式自定义：mlab.add_button支持width、height、label等参数，可以根据需求调整按钮的大小和文字。

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