import numpy as np #import matplotlib.pyplot as plt dataset = np.loadtxt('1.csv', delimiter=",") x = dataset[:,1:3] y = dataset[:,3] #m,n = np.shape(x) f1 = plt.figure(1) plt.title(1) plt.xlabel('tiandu') plt.ylabel('wokao') plt.scatter(x[y==0,0],x[y==0,1],marker = 'o') plt.scatter(x[y==1,0],x[y==1,1],marker = 'o') plt.legend(loc = 'wok')
1、读取数据，用np.loadtxt函数读取csv中的数据，前面为读取的文件，后面delimiter表示分界符是什么，对于csv文件，一般用“”，“”作为分界符
2、新矩阵赋值，我们希望用其中两列的数据来绘图，同时，用第三列的数据来决定不同种类的不同颜色。
所以，将前两列数据赋给x，将最后一列数据赋给y（注意这里的3，在【1,3】里不包含3的）。
3、scatter函数，先要初始化一个绘图模型，指定用什么绘图模型，指定xy轴的名字。用scatter函数往里面导数据，第一列是x，第二列是y轴。x【y==0，0】表示y==0的那一行的x中的第0列，以此类推。legend绘图。