plt.scatter()函数

函数用于生成一个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()