pyqt显示vtk代码

以下是一个简单的PyQt显示VTK的代码：

import vtk
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from vtk.qt.QVTKRenderWindowInteractor import QVTKRenderWindowInteractor

class VTKWidget(QWidget):
    def __init__(self, parent=None):
        super(VTKWidget, self).__init__(parent)

        self.vl = QVBoxLayout()
        self.setLayout(self.vl)

        self.vtkWidget = QVTKRenderWindowInteractor(self)
        self.vl.addWidget(self.vtkWidget)

        self.ren = vtk.vtkRenderer()
        self.vtkWidget.GetRenderWindow().AddRenderer(self.ren)

        self.iren = self.vtkWidget.GetRenderWindow().GetInteractor()

    def setRenderWindow(self, renderWindow):
        self.vtkWidget.SetRenderWindow(renderWindow)

if __name__ == "__main__":
    app = QApplication([])
    widget = VTKWidget()

    cylinder = vtk.vtkCylinderSource()
    cylinder.SetHeight(3.0)
    cylinder.SetRadius(1.0)
    cylinder.SetResolution(10)

    mapper = vtk.vtkPolyDataMapper()
    mapper.SetInputConnection(cylinder.GetOutputPort())

    actor = vtk.vtkActor()
    actor.SetMapper(mapper)

    widget.ren.AddActor(actor)

    widget.resize(600, 600)
    widget.show()

    widget.iren.Initialize()
    widget.iren.Start()

    app.exec_()

希望这可以帮助到你。