能否将Mayavi绘图嵌入PyQt5组件？PyQt5程序嵌入Mayavi曲面绘图可行吗？

当然没问题！把Mayavi的绘图嵌入PyQt5组件完全可行，而且掌握正确方法后实现起来很简单。我给你一步步拆解，再附上完整的可运行代码示例：

Mayavi专门提供了适配Qt框架的组件，我们可以用mayavi.core.ui.api中的MayaviQWidget来封装Mayavi的可视化场景，然后把这个QWidget像普通PyQt5组件一样添加到布局里就行。

• 先确保你安装了所需依赖：pyqt5、mayavi和numpy（用来生成示例数据），可以用pip安装：pip install pyqt5 mayavi numpy
• 导入必要的模块：PyQt5的窗口、布局组件，Mayavi的相关模块，还有numpy
• 创建PyQt5的主窗口或目标容器组件
• 初始化Mayavi的可视化场景，生成MayaviQWidget实例
• 在Mayavi场景中绘制你需要的曲面（这里用mlab的surf函数举例）
• 将MayaviQWidget添加到PyQt5的布局中
• 启动PyQt5的应用循环

import numpy as np
from PyQt5.QtWidgets import QApplication, QMainWindow, QWidget, QVBoxLayout
from mayavi.core.ui.api import MayaviQWidget
from mayavi import mlab

# 设置Mayavi使用Qt5后端，避免兼容性问题
mlab.options.backend = 'qt5'

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("Mayavi嵌入PyQt5曲面示例")
        self.setGeometry(100, 100, 800, 600)

        # 创建主容器和布局
        central_widget = QWidget()
        self.setCentralWidget(central_widget)
        layout = QVBoxLayout(central_widget)

        # 创建Mayavi的QWidget组件
        self.mayavi_widget = MayaviQWidget()
        layout.addWidget(self.mayavi_widget)

        # 在Mayavi组件中绘制曲面
        self.plot_custom_surface()

    def plot_custom_surface(self):
        # 生成示例曲面数据（替换成你自己的数据即可）
        x, y = np.mgrid[-3:3:100j, -3:3:100j]
        z = np.sin(x**2 + y**2)

        # 获取Mayavi的场景对象
        scene = self.mayavi_widget.scene

        # 绘制曲面
        mlab.surf(x, y, z, warp_scale="auto", figure=scene.mayavi_scene)
        # 调整视角让曲面展示更清晰
        mlab.view(azimuth=30, elevation=60, distance=10)

if __name__ == "__main__":
    app = QApplication([])
    window = MainWindow()
    window.show()
    app.exec_()

如果你的PyQt5应用已经有现成的组件，只需要把上述代码中创建MayaviQWidget并添加到布局的部分，整合到你的现有代码里就行：

1. 找到你要嵌入绘图的目标组件的布局
2. 实例化MayaviQWidget
3. 调用布局的addWidget()方法把这个组件加进去
4. 再调用自定义的绘图函数生成曲面即可

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