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 self,cls,decorator的理解
Jul 13 Python
python使用PyV8执行javascript代码示例分享
Dec 04 Python
归纳整理Python中的控制流语句的知识点
Apr 14 Python
Python实现将16进制字符串转化为ascii字符的方法分析
Jul 21 Python
Django Admin实现三级联动的示例代码(省市区)
Jun 22 Python
Python 使用Numpy对矩阵进行转置的方法
Jan 28 Python
pyinstaller打包单文件时--uac-admin选项不起作用怎么办
Apr 15 Python
python 3.8.3 安装配置图文教程
May 21 Python
Python logging模块原理解析及应用
Aug 13 Python
python语言time库和datetime库基本使用详解
Dec 25 Python
Pytorch中使用ImageFolder读取数据集时忽略特定文件
Mar 23 Python
Python必备技巧之字符数据操作详解
Mar 23 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
Yii框架参数化查询中IN查询只能查询一个的解决方法
2017/05/20 PHP
PHP 实现页面静态化的几种方法
2017/07/23 PHP
HTTP状态代码以及定义(解释)
2007/02/02 Javascript
javascript整除实现代码
2010/11/23 Javascript
图片动画横条广告带上下滚动的JS代码
2013/10/25 Javascript
前台js对象在后台转化java对象的问题探讨
2013/12/20 Javascript
浅析js预加载/延迟加载
2014/09/25 Javascript
Javascript快速排序算法详解
2014/12/03 Javascript
JavaScript中string转换成number介绍
2014/12/31 Javascript
详解JavaScript正则表达式中的global属性的使用
2015/06/16 Javascript
JS动态给对象添加事件的简单方法
2016/07/19 Javascript
javascript iframe跨域详解
2016/10/26 Javascript
解决nodejs中使用http请求返回值为html时乱码的问题
2017/02/18 NodeJs
详解Vue调用手机相机和相册以及上传
2019/05/05 Javascript
基于纯JS实现多张图片的懒加载Lazy过程解析
2019/10/14 Javascript
vue-router结合vuex实现用户权限控制功能
2019/11/14 Javascript
解决nuxt 自定义全局方法,全局属性,全局变量的问题
2020/11/05 Javascript
Python中用Decorator来简化元编程的教程
2015/04/13 Python
Django查找网站项目根目录和对正则表达式的支持
2015/07/15 Python
Python构建网页爬虫原理分析
2017/12/19 Python
Python实现的栈、队列、文件目录遍历操作示例
2019/05/06 Python
Python利用神经网络解决非线性回归问题实例详解
2019/07/19 Python
python3 下载网络图片代码实例
2019/08/27 Python
完美解决python针对hdfs上传和下载的问题
2020/06/05 Python
Python 爬虫性能相关总结
2020/08/03 Python
HTML5在线预览PDF的示例代码
2017/09/14 HTML / CSS
幼儿园教师奖惩制度
2014/02/01 职场文书
手机被没收检讨书
2014/02/22 职场文书
根叔历年演讲稿
2014/05/20 职场文书
文明工地标语
2014/06/16 职场文书
裁员通知
2015/04/25 职场文书
聚众斗殴罪辩护词
2015/05/21 职场文书
校园安全学习心得体会
2016/01/18 职场文书
2016年企业先进员工事迹材料
2016/02/25 职场文书
商业计划书之服装
2019/09/09 职场文书
python实现A*寻路算法
2021/06/13 Python