plt.contourf
时间: 2023-10-25 13:07:05 浏览: 48
`plt.contourf` 是 Matplotlib 库中的一个函数,用于绘制二维等高线图。它可以根据不同的数据值在二维平面上绘制出不同颜色的区域,形成类似于地形图的效果。
该函数的完整用法如下:
```python
plt.contourf(X, Y, Z, levels=None, cmap=None, alpha=None, origin=None, extent=None, …)
```
其中,参数 `X` 和 `Y` 分别表示二维平面上的 x 和 y 坐标。参数 `Z` 表示每个坐标点对应的数值。参数 `levels` 可以指定等高线的数量和高度,参数 `cmap` 可以指定颜色映射,参数 `alpha` 可以指定透明度,参数 `origin` 可以指定坐标轴方向,参数 `extent` 可以指定绘图区域的范围,等等。
例如,下面的代码演示了如何使用 `plt.contourf` 绘制一个简单的二维等高线图:
```python
import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2))
plt.contourf(X, Y, Z, levels=20, cmap='coolwarm')
plt.colorbar()
plt.show()
```
这段代码首先生成了一个 100x100 的二维网格,然后计算了每个网格点对应的数值。最后使用 `plt.contourf` 函数将这些数据绘制成等高线图,并使用 `coolwarm` 颜色映射来表示不同的数值。运行这段代码,会得到如下的图像:
![contourf](https://img-blog.csdnimg.cn/20211208132909514.png)
阅读全文