在加载了vtk的.stl模型上局部着色的方法

import vtk

# 创建vtkSTLReader对象
reader = vtk.vtkSTLReader()
# 设置.stl模型文件路径
reader.SetFileName("path/to/model.stl")
# 执行读取操作
reader.Update()

# 获取加载的模型数据
polydata = reader.GetOutput()

import random

# 获取模型顶点数
numPoints = polydata.GetNumberOfPoints()

# 创建颜色数组
colors = vtk.vtkUnsignedCharArray()
colors.SetNumberOfComponents(3)
colors.SetName("Colors")

# 为每个顶点分配随机颜色
for i in range(numPoints):
    color = [random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)]
    colors.InsertNextTuple(color)

# 将颜色数组与模型数据关联
polydata.GetPointData().SetScalars(colors)

# 创建vtkPolyDataMapper对象
mapper = vtk.vtkPolyDataMapper()
mapper.SetInputData(polydata)

# 创建vtkActor对象
actor = vtk.vtkActor()
actor.SetMapper(mapper)

# 创建vtkRenderer和vtkRenderWindow对象
renderer = vtk.vtkRenderer()
renderWindow = vtk.vtkRenderWindow()
renderWindow.AddRenderer(renderer)

# 将actor添加到renderer中
renderer.AddActor(actor)

# 创建vtkRenderWindowInteractor对象并启动交互
interactor = vtk.vtkRenderWindowInteractor()
interactor.SetRenderWindow(renderWindow)
interactor.Initialize()
interactor.Start()