python画三维函数图_使用Python+VTK实现三维模型的显示和切割（面绘制）

要画三维函数图，可以使用Python中的Matplotlib模块。以下是一个简单的例子：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

# 定义函数
def f(x, y):
    return np.sin(np.sqrt(x**2 + y**2))

# 生成数据
x = np.linspace(-5, 5, 50)
y = np.linspace(-5, 5, 50)
X, Y = np.meshgrid(x, y)
Z = f(X, Y)

# 绘制图形
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_surface(X, Y, Z)
plt.show()

如果要使用VTK来实现三维模型的显示和切割，可以参考下面的代码：

import vtk

# 读取STL文件
reader = vtk.vtkSTLReader()
reader.SetFileName("model.stl")
reader.Update()

# 显示模型
mapper = vtk.vtkPolyDataMapper()
mapper.SetInputConnection(reader.GetOutputPort())

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

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

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

# 切割模型
plane = vtk.vtkPlane()
plane.SetOrigin(0, 0, 0)
plane.SetNormal(1, 0, 0)

cutter = vtk.vtkCutter()
cutter.SetInputConnection(reader.GetOutputPort())
cutter.SetCutFunction(plane)
cutter.Update()

cutterMapper = vtk.vtkPolyDataMapper()
cutterMapper.SetInputConnection(cutter.GetOutputPort())

cutterActor = vtk.vtkActor()
cutterActor.SetMapper(cutterMapper)

# 显示切割后的模型
renderer2 = vtk.vtkRenderer()
renderer2.AddActor(cutterActor)

renderWindow2 = vtk.vtkRenderWindow()
renderWindow2.AddRenderer(renderer2)

# 显示窗口
iren = vtk.vtkRenderWindowInteractor()
iren.SetRenderWindow(renderWindow)
iren.Initialize()
iren.Start()

这个例子中，我们首先读取一个STL文件，然后显示整个模型。接着，我们定义了一个切割平面，使用vtkCutter对模型进行切割，并显示切割后的模型。最后，我们使用vtkRenderWindowInteractor来显示窗口，并启动事件循环。