使用Python构建地质三维可视化模型有多种方法，以下是一些示例： ### 绘制3D切片图 通过`Visualizer`类的`plot_3d_slice`方法可以绘制地质相关的3D切片图，适用于如地下水流动模拟等场景。代码示例如下： ```python import matplotlib.pyplot as plt class Visualizer: @staticmethod def plot_3d_slice(solution, model): """绘制3D切片图""" fig = plt.figure(figsize=(12, 8)) ax = fig.add_subplot(111, projection='3d') # 获取网格坐标和值 coords = model.mesh.coordinates() values = solution.compute_vertex_values(model.mesh) # 绘制散点图 sc = ax.scatter(coords[:,0], coords[:,1], coords[:,2], c=values, cmap='viridis', alpha=0.6) plt.colorbar(sc, label='Hydraulic Head (m)') ax.set_xlabel('X (m)') ax.set_ylabel('Y (m)') ax.set_zlabel('Depth (m)') plt.title('3D Groundwater Flow Simulation') plt.show() ``` 此方法可展示地质模型中不同位置的水力水头分布情况 [^1]。 ### 绘制3D曲面图 使用`matplotlib`的`plot_surface`方法可以绘制3D曲面图，适用于地形建模和数学函数可视化，如莫比乌斯带、地形高程图以及像`sin(√(x² + y²))`这类数学函数的波纹表面展示。代码示例如下： ```python import numpy as np import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D # 定义函数 def fun(x, y): return np.sin(np.sqrt(x**2 + y**2)) # 生成网格 x = np.linspace(-5, 5, 100) y = np.linspace(-5, 5, 100) X, Y = np.meshgrid(x, y) Z = fun(X, Y) fig = plt.figure() ax = fig.add_subplot(111, projection='3d') ax.plot_surface(X, Y, Z, cmap='cool', alpha=0.8) plt.show() ``` 该方法支持透明度（`alpha`）和颜色映射（`cmap`），可增强视觉效果，还能叠加等高线投影 [^2]。 ### 使用VTK图形库 VTK是跨平台三维可视化与图形库，可用于创建复杂的三维可视化模型。以下是创建一个球体的示例： ```python import vtk # 创建一个球体 sphere = vtk.vtkSphereSource() sphere.SetRadius(1.0) # 创建映射器和演员 mapper = vtk.vtkPolyDataMapper() mapper.SetInputConnection(sphere.GetOutputPort()) actor = vtk.vtkActor() actor.SetMapper(mapper) # 设置渲染窗口 renderer = vtk.vtkRenderer() renderer.AddActor(actor) renderWindow = vtk.vtkRenderWindow() renderWindow.AddRenderer(renderer) # 启动交互式窗口 interactor = vtk.vtkRenderWindowInteractor() interactor.SetRenderWindow(renderWindow) interactor.Start() ``` VTK可以构建更专业、复杂的三维地质模型 [^4]。