plt.contourf用法
时间: 2024-05-07 17:21:52 浏览: 170
用matplotlib画等高线图详解
`plt.contourf` 是 Matplotlib 库中的一个函数,用于绘制等高线图,即在二维平面上绘制出不同高度的区域。它的基本用法如下:
```python
plt.contourf(X, Y, Z, levels, **kwargs)
```
其中,`X` 和 `Y` 表示要绘制的网格点坐标,`Z` 表示对应的高度值,`levels` 表示等高线的数量或高度值。`**kwargs` 中包含一些可选参数,如 `cmap` 表示颜色映射,`alpha` 表示透明度等。
下面是一个简单的例子:
```python
import numpy as np
import matplotlib.pyplot as plt
# 生成数据
x = np.linspace(-2, 2, 200)
y = np.linspace(-2, 2, 200)
X, Y = np.meshgrid(x, y)
Z = X**2 + Y**2
# 绘制等高线图
plt.contourf(X, Y, Z, levels=10, cmap='hot')
# 显示图像
plt.show()
```
运行这段代码可以看到一个简单的等高线图,其中红色表示高度较高的区域,黄色表示高度较低的区域。你可以调整 `levels` 参数以改变等高线的数量或高度值,也可以使用其他的颜色映射来改变颜色。
阅读全文