基于numpy的绘图

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绘图。

转载于:https://www.cnblogs.com/baochen/p/9036149.html