plt.contourf
plt.contourf用来画出不同分类的边界线
1、生成数据点
x = np.arange(-5,5,1) y = np.arange(0,20,2) xx,yy = np.meshgrid(x,y)
2、对不同类进行标记
z = np.square(xx) - yy >0
3、生成边界图
如果点去的比较密集:可以看到完整的抛物线形状
4、完整代码
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(-5,5,0.1)
y = np.arange(0,20,0.2)
xx,yy = np.meshgrid(x,y)
z = np.square(xx) - yy >0
print(z)
plt.contourf(xx,yy,z,cmap="cool")
plt.scatter(xx,yy,c = z)
plt.show()