【Python Matplotlib 教程】第23课时-Matplotlib 等高线图

Matplotlib 等高线图

等高线图（有时称为水平面图）是一种将三维表面绘制在二维平面上的方法。它在y轴上绘制了两个预测变量X和Y，而响应变量Z则以等高线的形式呈现。这些等高线有时被称为z-切片或等响应值。

如果您想观察变量Z随着两个输入变量X和Y的变化情况，使得Z = f（X，Y），则等高线图很合适。二元函数的等值线或等值线是函数取常数值的曲线。

独立变量x和y通常限制在一个称为meshgrid的规则网格上。numpy.meshgrid将一组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()

Python