plt.scatter()函数

最新推荐文章于 2024-09-04 10:31:20 发布

泮城～烟沙

最新推荐文章于 2024-09-04 10:31:20 发布

阅读量976

点赞数

分类专栏：深度学习文章标签： python 人工智能信息可视化

本文链接：https://blog.csdn.net/weixin_54606863/article/details/128441511

版权

深度学习专栏收录该内容

4 篇文章 0 订阅

订阅专栏

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