在python中将vtk UnstrucutredGrid写入文件

import vtk

grid = vtk.vtkUnstructuredGrid()

# 创建一个vtk数据单元（四边形）
quad = vtk.vtkQuad()
quad.GetPointIds().SetId(0, 0)  # 设置第一个顶点的索引
quad.GetPointIds().SetId(1, 1)  # 设置第二个顶点的索引
quad.GetPointIds().SetId(2, 2)  # 设置第三个顶点的索引
quad.GetPointIds().SetId(3, 3)  # 设置第四个顶点的索引

# 将vtk数据单元添加到UnstructuredGrid中
grid.InsertNextCell(quad.GetCellType(), quad.GetPointIds())

# 创建一个vtk数据数组（标量值）
scalars = vtk.vtkFloatArray()
scalars.SetNumberOfComponents(1)
scalars.SetName("Scalars")

# 设置数据数组的属性值
scalars.InsertNextValue(1.0)
scalars.InsertNextValue(2.0)
scalars.InsertNextValue(3.0)
scalars.InsertNextValue(4.0)

# 将vtk数据数组添加到UnstructuredGrid中
grid.GetPointData().SetScalars(scalars)

# 创建一个vtk数据写入器
writer = vtk.vtkUnstructuredGridWriter()
writer.SetFileName("output.vtk")

# 将UnstructuredGrid对象传递给数据写入器，并执行写入操作
writer.SetInputData(grid)
writer.Write()