使用contourf创建一个等高线集合

要使用 `contourf` 函数创建等高线集合，您需要提供 X 和 Y 值的网格，以及对应于每个网格点的 Z 值，其中 Z 值是用于绘制等高线的高度值。以下是一个简单的示例代码：

import numpy as np
import matplotlib.pyplot as plt

# 生成 X 和 Y 值的网格
x = np.linspace(-3, 3, 100)
y = np.linspace(-3, 3, 100)
X, Y = np.meshgrid(x, y)

# 计算每个网格点的高度值 Z
Z = np.sin(np.sqrt(X**2 + Y**2))

# 绘制等高线集合
plt.contourf(X, Y, Z, cmap='viridis')

# 显示结果
plt.colorbar()
plt.show()

在这个例子中，我们首先使用 `np.linspace` 函数生成 X 和 Y 值的网格，然后使用 `np.meshgrid` 函数将它们组合成一个网格。接下来，我们计算每个网格点的高度值 Z，这里使用了一个简单的函数 `np.sin(np.sqrt(X**2 + Y**2))`。最后，我们使用 `contourf` 函数绘制等高线集合，并使用 `cmap='viridis'` 参数指定颜色映射。 `plt.colorbar()` 函数用于创建 colorbar，以便我们可以查看颜色映射。

相关问题

PYTHON 让cad绘制三角网和等高线scipy.spatial

Python 的 `scipy` 库中的 `scipy.spatial` 模块主要用于处理各种空间数据结构，包括点集、线段集合等。如果要在 CAD 环境中利用 Python 绘制三角网 (Triangulation) 和等高线 (Contours)，你可以结合使用 `matplotlib` 和 `trimesh` 这两个库。

`matplotlib` 是用于数据可视化的强大工具，可以创建各种图表，包括三维图形。而 `trimesh` 则专注于三维几何体的操作，尤其是构建和渲染复杂的三角形网格。

以下是简要步骤：

1. 安装所需库：

   pip install matplotlib trimesh scipy

2. 导入所需的模块：

   import matplotlib.pyplot as plt
   from matplotlib.tri import Triangulation
   import numpy as np
   import trimesh

3. 创建等高线数据（例如从数值数组得到）：

   x, y = np.random.rand(2, 100)  # 假设这是你的二维数据
   z = np.sin(x**2 + y**2)  # 示例等高线函数

   tri = Triangulation(x, y)

4. 绘制三角网：

   mesh = trimesh.Trimesh(vertices=tri.points, faces=tri.triangles)
   ax = plt.figure().add_subplot(projection='3d')
   ax.plot_trisurf(x, y, z, triangles=mesh.faces, linewidth=0.2)

5. 绘制等高线：

   contourf = ax.tricontourf(tri, z)
   plt.colorbar(contourf)

6. 显示图形：

   plt.show()