python scatter散点图用循环分类法加图例
作者:xiaobaicai4552 时间:2021-07-26 01:44:01
本文实例为大家分享了python scatter散点图用循环分类法加图例,供大家参考,具体内容如下
import matplotlib.pyplot as plt
import kNN
plt.rcParams['font.sans-serif']=['Simhei']
plt.rcParams['axes.unicode_minus']=False
datingDataMat, datingLabels = kNN.file2matrix('datingTestSet2.txt')
plt.figure()
type1_x = [] #一共有3类,所以定义3个空列表准备接受数据
type1_y = []
type2_x = []
type2_y = []
type3_x = []
type3_y = []
for i in range(len(datingLabels)): #1000组数据,i循环1000次
if datingLabels[i] == '1': #根据标签进行数据分类,注意标签此时是字符串
type1_x.append(datingDataMat[i][0]) #取的是样本数据的第一列特征和第二列特征
type1_y.append(datingDataMat[i][1])
if datingLabels[i] == '2':
type2_x.append(datingDataMat[i][0])
type2_y.append(datingDataMat[i][1])
if datingLabels[i] == '3':
type3_x.append(datingDataMat[i][0])
type3_y.append(datingDataMat[i][1])
plt.scatter(type1_x, type1_y, s=20, c='r', label='不喜欢')
plt.scatter(type2_x, type2_y, s=40, c='b', label='魅力一般')
plt.scatter(type3_x, type3_y, s=60, c='k', label='极具魅力')
plt.legend()
plt.show()
用面向对象的写法:
import matplotlib.pyplot as plt
import kNN
plt.rcParams['font.sans-serif']=['Simhei']
plt.rcParams['axes.unicode_minus']=False
datingDataMat, datingLabels = kNN.file2matrix('datingTestSet2.txt')
plt.figure()
axes = plt.subplot(111)
type1_x = []
type1_y = []
type2_x = []
type2_y = []
type3_x = []
type3_y = []
for i in range(len(datingLabels)):
if datingLabels[i] == '1':
type1_x.append(datingDataMat[i][0])
type1_y.append(datingDataMat[i][1])
if datingLabels[i] == '2':
type2_x.append(datingDataMat[i][0])
type2_y.append(datingDataMat[i][1])
if datingLabels[i] == '3':
type3_x.append(datingDataMat[i][0])
type3_y.append(datingDataMat[i][1])
type1 = axes.scatter(type1_x, type1_y, s=20, c='r')
type2 = axes.scatter(type2_x, type2_y, s=40, c='b')
type3 = axes.scatter(type3_x, type3_y, s=60, c='k')
plt.legend((type1, type2, type3), ('不喜欢', '魅力一般', '极具魅力'))
plt.show()
来源:https://blog.csdn.net/xiaobaicai4552/article/details/79069207
标签:python,scatter,散点图
0
投稿
猜你喜欢
Python调用ChatGPT制作基于Tkinter的桌面时钟
2023-05-11 08:23:23
Python入门教程(八)PythonCasting用法
2021-11-14 02:20:41
Python tkinter制作单机五子棋游戏
2021-08-01 03:31:43
520使用Python实现“我爱你”表白
2022-07-16 05:25:47
Python微信企业号开发之回调模式接收微信端客户端发送消息及被动返回消息示例
2023-09-20 13:29:27
PHP解决高并发问题(opcache)
2023-11-07 23:17:49
js获取 type=radio 值的方法
2024-04-29 13:18:38
xheditor所见即所得文本编辑器(代码高亮显示修改)
2022-04-17 02:20:25
Python xlwt设置excel单元格字体及格式
2023-03-10 06:04:39
gearman + mysql方式实现持久化操作示例
2024-01-16 14:00:22
关于从MySQL转向ADODB的方法
2011-06-02 12:05:00
深入浅析python中的多进程、多线程、协程
2022-06-05 15:59:27
使用python进行波形及频谱绘制的方法
2023-02-07 02:48:58
Python全景系列之数据类型大盘点
2021-01-10 15:29:19
JavaScript DOM节点操作实例小结(新建,删除HTML元素)
2024-04-17 10:37:57
python控制nao机器人身体动作实例详解
2023-08-26 11:33:17
ASP自动清除ACCESS数据库的日文字符
2007-11-28 17:40:00
数据库备份过程中经常遇到的九种情况
2008-12-26 16:38:00
TypeScript工具类 Partial 和 Required 的场景分析
2024-04-29 13:14:45
Python爬虫工具requests-html使用解析
2021-10-23 02:20:27