用python matplotlib画一个基本的分类图

在juputer编辑器中完成

首先要运行%matplotlib inline

然后这段代码

import numpy as np
import matplotlib.pyplot as plt
plt.style.use('ggplot')
#setting plt
# 确定横纵坐标范围
plt.xlim(0,8)
plt.ylim(0,8)
# 确定横纵轴标题
plt.xlabel('x1')
plt.ylabel('x2')
# 确定该图的标题
plt.title('Discriminant interface')
# 把想画的散点按照横纵坐标输入进来
x=[0,2,2,0]
y=[0,0,2,2]
# bs是点的形状
plt.plot(x,y,'bs',color='red',label='w1')

x=[4,6,6,4]
y=[4,4,6,6]
# g^是点的形状
plt.plot(x,y,'g^',color='green',label='w2')
# 再画一条分类的线,线的范围是0-8
x= np.linspace(0,8)
plt.plot(x,6-x)
# 把点的标注协商
plt.legend(loc='upper left', shadow=True, fontsize='x-large')
plt.grid(True)
plt.savefig('E:/test.jpg')
plt.show()

代码运行后图像如图所示
运行后图像

缺的包用pip安装即可。

猜你喜欢

转载自blog.csdn.net/weixin_42379358/article/details/82831259