Matplotlib - 等高线图(Contour)

等高线图(有时称为“水平图”)是一种在二维平面上显示三维表面的方法。它在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()