python scatter散点图用循环分类法加图例


Posted in Python onMarch 19, 2019

本文实例为大家分享了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()

python scatter散点图用循环分类法加图例

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
如何运行Python程序的方法
Apr 21 Python
Linux下使用python调用top命令获得CPU利用率
Mar 10 Python
打包发布Python模块的方法详解
Sep 18 Python
python算法表示概念扫盲教程
Apr 13 Python
解决python3中解压zip文件是文件名乱码的问题
Mar 22 Python
Python实现去除列表中重复元素的方法小结【4种方法】
Apr 27 Python
详解PyTorch批训练及优化器比较
Apr 28 Python
如何利用pygame实现简单的五子棋游戏
Dec 29 Python
PyTorch和Keras计算模型参数的例子
Jan 02 Python
Tensorflow卷积实现原理+手写python代码实现卷积教程
May 22 Python
浅谈python量化 双均线策略(金叉死叉)
Jun 03 Python
python的列表生成式,生成器和generator对象你了解吗
Mar 16 Python
python使用matplotlib画柱状图、散点图
Mar 18 #Python
matplotlib实现区域颜色填充
Mar 18 #Python
Python函数定义及传参方式详解(4种)
Mar 18 #Python
Python中super函数用法实例分析
Mar 18 #Python
Python操作MySQL数据库的两种方式实例分析【pymysql和pandas】
Mar 18 #Python
python+pyqt5编写md5生成器
Mar 18 #Python
python+pyqt5实现图片批量缩放工具
Mar 18 #Python
You might like
德劲1102收音机的打理维修案例
2021/03/02 无线电
深入php 正则表达式的学习探讨
2013/06/06 PHP
PHP面向对象教程之自定义类
2014/06/10 PHP
通过php删除xml文档内容的方法
2015/01/23 PHP
PHP常用处理静态操作类
2015/04/03 PHP
记录一次排查PHP脚本执行卡住的问题
2016/12/27 PHP
xml和web特殊字符
2009/04/28 Javascript
实现动画效果核心方式的js代码
2013/09/27 Javascript
解析URI与URL之间的区别与联系
2013/11/22 Javascript
js如何设置在iframe框架中指定div不显示
2013/12/04 Javascript
禁止IE用右键的JS代码
2013/12/30 Javascript
js创建一个input数组并绑定click事件的方法
2014/06/12 Javascript
js控制再次点击按钮之间的间隔时间可防止重复提交
2014/08/01 Javascript
React.js入门实例教程之创建hello world 的5种方式
2016/05/11 Javascript
浅谈NodeJs之数据库异常处理
2017/10/25 NodeJs
three.js实现3D影院的原理的代码分析
2017/12/18 Javascript
微信小程序之批量上传并压缩图片的实例代码
2018/07/05 Javascript
js实现随机抽奖
2020/03/19 Javascript
js实现页面导航层级指示效果
2020/08/25 Javascript
js实现石头剪刀布游戏
2020/10/11 Javascript
[56:58]VP vs Optic 2018国际邀请赛小组赛BO2 第一场 8.16
2018/08/17 DOTA
python验证码识别教程之利用滴水算法分割图片
2018/06/05 Python
Python openpyxl 遍历所有sheet 查找特定字符串的方法
2018/12/10 Python
Python实现的在特定目录下导入模块功能分析
2019/02/11 Python
itchat-python搭建微信机器人(附示例)
2019/06/11 Python
Python使用Slider组件实现调整曲线参数功能示例
2019/09/06 Python
Django 限制访问频率的思路详解
2019/12/24 Python
Python文件操作模拟用户登陆代码实例
2020/06/09 Python
Python 实现进度条的六种方式
2021/01/06 Python
html5 Canvas画图教程(2)—画直线与设置线条的样式如颜色/端点/交汇点
2013/01/09 HTML / CSS
世界领先的26岁以下学生和青少年旅行预订网站:StudentUniverse
2018/07/01 全球购物
美国温暖商店:The Warming Store
2018/12/15 全球购物
英国DIY汽车维修配件网站:DIY Car Service Parts
2019/08/30 全球购物
JD Sports澳洲官网:英国领先的运动鞋和运动时尚零售商
2020/02/15 全球购物
取保候审保证书
2014/04/30 职场文书
邀请函怎么写
2015/01/30 职场文书