如何将3DNumpy数组转换为灰度vtkImageData在pyQT中使用

import vtk
import numpy as np
from vtk.qt.QVTKRenderWindowInteractor import QVTKRenderWindowInteractor

imageData = vtk.vtkImageData()
imageData.SetDimensions(shape[2], shape[1], shape[0])
imageData.SetSpacing(1, 1, 1)
imageData.SetOrigin(0, 0, 0)
imageData.AllocateScalars(vtk.VTK_UNSIGNED_CHAR, 1)

array = np.transpose(array, (2, 1, 0))  # 调整数组维度顺序
array = np.ascontiguousarray(array)  # 确保数组连续性
vtk_array = vtk.util.numpy_support.numpy_to_vtk(array.ravel(), deep=True, array_type=vtk.VTK_UNSIGNED_CHAR)
imageData.GetPointData().SetScalars(vtk_array)

imageActor = vtk.vtkImageActor()
imageActor.SetInputData(imageData)

renderer = vtk.vtkRenderer()
renderer.AddActor(imageActor)

renderWindow = vtk.vtkRenderWindow()
renderWindow.AddRenderer(renderer)

renderWindowInteractor = QVTKRenderWindowInteractor()
renderWindowInteractor.SetRenderWindow(renderWindow)

# 假设pyQT的主窗口为mainWindow
vtkWidget = renderWindowInteractor.GetRenderWindow().GetInteractor()
vtkWidget.SetParent(mainWindow)
vtkWidget.setGeometry(0, 0, width, height)  # 设置vtk窗口的位置和大小
vtkWidget.show()