基于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

你可能感兴趣的:(python,人工智能)