等高线图(有时称为“水平图”)是一种在二维平面上显示三维表面的方法。它在y轴上绘制两个预测变量X Y,并以轮廓绘制响应变量Z。
如果您想查看Z随两个输入X和Y的变化,等高线图是合适的,那么Z=f(X,Y)。
Matplotlib API包含分别绘制轮廓线和填充轮廓的Contour()和Contourf()函数。这两个函数都需要三个参数x,y和z。
import numpy as np import matplotlib.pyplot as plt xlist=np.linspace(-3.0, 3.0, 100) ylist=np.linspace(-3.0, 3.0, 100) X, Y=np.meshgrid(xlist, ylist) Z=np.sqrt(X**2 + Y**2) fig,ax=plt.subplots(1,1) cp=ax.contourf(X, Y, Z) fig.colorbar(cp) # Add a colorbar to a plot ax.set_title('Filled Contours Plot') #ax.set_xlabel('x (cm)') ax.set_ylabel('y (cm)') plt.show()
![Filled Colours Plot](http://dev-img.mos.moduyun.com/20231017/24c66efe-8a2b-4194-be14-8a48e17546e6.jpg)
参考链接
https://www.learnfk.com/matplotlib/matplotlib-contour-plot.html