在Python中,使用`matplotlib`库可以绘制等高线图。以下是使用`contourf`函数绘制等高线图的步骤和示例代码:
1. 导入必要的库:
```python
import numpy as np
import matplotlib.pyplot as plt
2. 定义一个函数来计算高度值(这里以`fun`函数为例):
```python
def fun(x, y):
return (1 - x / 2 + x 2 + y 3) * np.exp(-x 2 - y 2)
3. 设置背景色和生成网格数据:
```python
plt.figure(facecolor="66ff33")
x = np.linspace(-3, 3, 256)
y = np.linspace(-3, 3, 256)
X, Y = np.meshgrid(x, y)
4. 计算高度值并填充等高线:
```python
Z = fun(X, Y)
plt.contourf(X, Y, Z, levels=20, cmap="GnBu", alpha=0.5)
5. 显示图表:
```python
plt.colorbar()
plt.show()
以上代码将生成一个填充的等高线图,其中`contourf`函数用于填充等高线之间的区域,`levels`参数用于确定等高线的数量和位置,`cmap`参数用于设置颜色映射。
如果你需要绘制轮廓线而不是填充区域,可以使用`contour`函数。
希望这能帮助你绘制出等高线图!