ax.contourf举例
时间: 2023-09-22 19:06:38 浏览: 219
用matplotlib画等高线图详解
当你想要绘制一个二元函数 $f(x, y) = \sin(x) + \cos(y)$ 的等高线填充图时,可以使用ax.contourf()函数。以下是一个例子:
```python
import numpy as np
import matplotlib.pyplot as plt
# 生成数据
x = np.linspace(-np.pi, np.pi, 100)
y = np.linspace(-np.pi, np.pi, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(X) + np.cos(Y)
# 绘图
fig, ax = plt.subplots()
contour = ax.contourf(X, Y, Z, cmap='coolwarm')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_title('Contour plot of f(x,y)=sin(x)+cos(y)')
plt.colorbar(contour)
plt.show()
```
在这个例子中,我们首先生成了横纵坐标的网格,然后计算出每个网格的高度(或函数值),存储在Z中。我们使用ax.contourf()函数将X、Y、Z作为参数传入,绘制了等高线填充图。我们还使用cmap参数指定了颜色映射为'coolwarm',并使用colorbar()函数添加了颜色条。最后,我们添加了x、y轴标签和标题,并显示了图形。
阅读全文