等高线或者密度图是我们常用图形, Matplotlib提供三个函数来供我们使用:
- plt.contour: 等高线
- plt.contourf: 自带填充色
- plt.imshow: 显示图形
#准备环境
import matplotlib.pyplot as plt
import numpy as np
# 设置⻛格
plt.style.use('seaborn-whitegrid')
contour
我们需要⼀个三维函数,z=f(x,y)来演示等高线图,按照下⾯函数来进行生成。
contour创建需要至少三个参数,x,y和z,其中x,y我们可以用横轴纵轴表示,z用等高线来表示就可以。当只有⼀个颜色的图形是,虚线表示负值,实现部分表示正值。
我们使用meshgrid来从⼀维数据构成⼆维网格数据。
def f(x, y):
return np.sin(x) ** 10 + np.cos(10 + y*x) * np.cos(x)
x =