contourf函数用法
时间: 2023-07-26 16:46:15 浏览: 97
Matplotlib函数大全(中文版)
5星 · 资源好评率100%
contourf函数是Matplotlib中的一个函数,用于绘制等高线图并填充颜色。
其基本用法为:
```python
contourf(X,Y,Z)
```
其中,X和Y是数据的网格矩阵,Z是数据的值矩阵。通常,我们可以使用numpy的meshgrid函数生成网格矩阵。
举个例子,如果我们想要绘制二元函数 $f(x,y)=\sin(x)+\cos(y)$ 的等高线图,可以这样写:
```python
import numpy as np
import matplotlib.pyplot as plt
# 定义二元函数
def f(x,y):
return np.sin(x) + np.cos(y)
# 生成网格矩阵
x = np.linspace(-np.pi, np.pi, 100)
y = np.linspace(-np.pi, np.pi, 100)
X, Y = np.meshgrid(x, y)
# 计算值矩阵
Z = f(X,Y)
# 绘制等高线图并填充颜色
plt.contourf(X, Y, Z)
# 显示图像
plt.show()
```
运行该代码,就可以得到一个等高线图并填充颜色的图像。如果需要修改填充颜色的样式,可以设置参数cmap,例如:
```python
plt.contourf(X, Y, Z, cmap='coolwarm')
```
这样就可以使用颜色映射表coolwarm来填充颜色了。
阅读全文