plt.contourf用法
时间: 2023-06-12 20:08:24 浏览: 121
matplotlib函数汇总.rar
`plt.contourf`是一个Matplotlib函数,用于绘制等高线图。
函数的语法为:`plt.contourf(X, Y, Z, levels, **kwargs)`
其中,X和Y是表示网格点的数组,Z是这些点的高度值,levels是要绘制的等高线的高度值。kwargs是其他参数,例如颜色、线型等。
下面是一个简单的例子,演示如何使用`plt.contourf`绘制一个简单的等高线图:
```python
import numpy as np
import matplotlib.pyplot as plt
# 生成数据
x = np.linspace(-3, 3, 301)
y = np.linspace(-3, 3, 301)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2))
# 绘制等高线图
plt.contourf(X, Y, Z, levels=20, cmap='jet')
plt.colorbar()
plt.show()
```
这个例子中,我们首先生成了一组数据,然后使用`np.meshgrid`函数将这些数据转换成网格点的数组。接着,使用`np.sin`函数计算每个点的高度值。最后,调用`plt.contourf`函数绘制等高线图,并使用`levels`参数指定绘制的等高线的数量,使用`cmap`参数指定颜色映射,使用`colorbar`函数添加颜色条。
阅读全文