函数用于生成一个scatter散点图。
plt.scatter(X[arg,0], X[arg,1], s=100,c='b',marker='+',label='data1')
参数:
x,y:绘制散点图的数据点
s:大小
c:颜色
marker:标记的样式,默认“o”
label:标签名
import sklearn.datasets #引入数据集
import torch
import numpy as np
import matplotlib.pyplot as plt
np.random.seed(0) #设置随机数种子
X, Y = sklearn.datasets.make_moons(200,noise=0.2) #生成2组半圆形数据
arg = np.squeeze(np.argwhere(Y==0)) #获取第1组数据索引
arg2 = np.squeeze(np.argwhere(Y==1))#获取第2组数据索引
plt.title("moons data")
plt.scatter(X[arg,0], X[arg,1], s=100,c='b',marker='+',label='data1')
plt.scatter(X[arg2,0], X[arg2,1],s=40, c='r',marker='o',label='data2')
plt.legend()#图例
plt.show()