函数原型
contourf([X, Y,] Z, [levels], **kwargs)contour([X, Y,] Z, [levels], **kwargs)
函数说明
contour函数绘制三维图像到二维图像所对应的等高线,而contourf函数也是绘制等高线,只不过contourf是带填充的等高线。这两个函数的参数都是一样的。
参数X、Y表示等高线的坐标,Z表示等高线的高度,也就是坐标点(x, y)对应的高度h。如果X的长度为N,Y的长度为M,那么Z的形状为(N,M)。参数levels表示如果为一个整数表示等高线的条数,如果为一个数组表示等高线的位置。
函数使用
import matplotlib.pyplot as pltimport numpy as npplt.figure()plt.subplot()x, y = np.arange(-3, 4), np.arange(-3, 4)plt.xlim(np.min(x), np.max(x))plt.ylim(np.min(y), np.max(y))plt.xticks(x)plt.yticks(y)plt.xlabel("x_axis")plt.ylabel("y_axis")# 等高线的x、y坐标X, Y = np.meshgrid(np.linspace(-3, 3, 50), np.linspace(-3, 3, 50))# 等高线为一个圆Z = X ** 2 + Y ** 2# 画出高度为2、5、8、10的等高线C = plt.contour(X, Y, Z, levels=[2, 5, 8, 10])plt.clabel(C, inline=True, fontsize=10)plt.show()