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 创建子进程模块subprocess详解
Apr 08 Python
Python去除字符串两端空格的方法
May 21 Python
基于Python如何使用AIML搭建聊天机器人
Jan 27 Python
Fabric 应用案例
Aug 28 Python
python入门教程之识别验证码
Mar 04 Python
python记录程序运行时间的三种方法
Jul 14 Python
python扫描proxy并获取可用代理ip的实例
Aug 07 Python
python批量修改图片尺寸,并保存指定路径的实现方法
Jul 04 Python
Django中reverse反转并且传递参数的方法
Aug 06 Python
Python3开发实例之非关系型图数据库Neo4j安装方法及Python3连接操作Neo4j方法实例
Mar 18 Python
python 生成器需注意的小问题
Sep 29 Python
Python 全局空间和局部空间
Apr 06 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
PHP 事务处理数据实现代码
2010/05/13 PHP
rrmdir php中递归删除目录及目录下的文件
2011/05/15 PHP
ThinkPHP使用smarty模板引擎的方法
2014/07/01 PHP
PHP缓存机制Output Control详解
2014/07/14 PHP
为PHP安装imagick时出现Cannot locate header file MagickWand.h错误的解决方法
2014/11/03 PHP
PHP实现时间比较和时间差计算的方法示例
2017/07/24 PHP
php+mysql开发的最简单在线题库(在线做题系统)完整案例
2019/03/30 PHP
jquery ajax执行后台方法
2010/03/18 Javascript
收集的一些Array及String原型对象的扩展实现代码
2010/12/05 Javascript
jquery中动态效果小结
2010/12/16 Javascript
Javascript面向对象设计一 工厂模式
2011/12/20 Javascript
jQuery插件实现表格隔行换色且感应鼠标高亮行变色
2013/09/22 Javascript
贴近用户体验的Jquery日期、时间选择插件
2015/08/19 Javascript
JS数组返回去重后数据的方法解析
2017/01/03 Javascript
深入讲解xhr(XMLHttpRequest)/jsonp请求之abort
2017/07/26 Javascript
Javascript 严格模式use strict详解
2017/09/16 Javascript
微信小程序实现翻牌抽奖动画
2020/09/21 Javascript
python 用下标截取字符串的实例
2018/12/25 Python
python实现定时压缩指定文件夹发送邮件
2020/12/22 Python
Django中多种重定向方法使用详解
2019/07/17 Python
keras分类模型中的输入数据与标签的维度实例
2020/07/03 Python
Python如何进行时间处理
2020/08/06 Python
Python读取pdf表格写入excel的方法
2021/01/22 Python
浅析HTML5中header标签的用法
2016/06/24 HTML / CSS
世界上最大的皮肤科医生拥有和经营的美容网站:LovelySkin
2021/01/03 全球购物
口腔医学技术应届生求职信
2013/11/09 职场文书
我的求职计划书
2014/01/10 职场文书
应届毕业生如何写求职信
2014/02/16 职场文书
部队万能检讨书
2014/02/20 职场文书
房地产促销活动方案
2014/03/01 职场文书
文员求职信
2014/07/15 职场文书
党的群众路线对照检查材料(个人)
2014/09/24 职场文书
春秋淹城导游词
2015/02/11 职场文书
eclipse创建项目没有dynamic web的解决方法
2021/06/24 Java/Android
Jackson 反序列化时实现大小写不敏感设置
2021/06/29 Java/Android
Python中文纠错的简单实现
2021/07/07 Python