在【欠拟合、过拟合(L1正则化、L2正则化)】这篇文章里面,我们用了一个简单的等高线来帮助大家理解问题,那么有同学会问:我怎么才能画出高级的等高线的图呢?
今天我们就来学习一下,python如何画出等高线的,这里主要用到matplotlib.pyplot包的contour、contourf
contour:矩阵的等高线
contourf:填充的等高线
大家可能还不是很了解这两个有什么区别,那么我们就简单的举个例子来看看,这两个函数都是怎么画出等高线的
1、给出数据
首先我们给出一组数据X,Y,Z,其中
###### 数据X,Y,Z
x = np.arange(-10, 10, 0.01) # 步长为0.01,即每隔0.01取一个点
y = np.arange(-10, 10, 0.01) # 步长为0.01,即每隔0.01取一个点
X, Y = np.meshgrid(x, y) # 将原始数据变成网格数据形式
Z = X**2