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的即时标记项目练习笔记
Sep 18 Python
Python使用matplotlib绘制动画的方法
May 20 Python
python图像处理之反色实现方法
May 30 Python
Python实现字符串格式化的方法小结
Feb 20 Python
python3利用Dlib19.7实现人脸68个特征点标定
Feb 26 Python
通过Python 接口使用OpenCV的方法
Apr 02 Python
python requests爬取高德地图数据的实例
Nov 10 Python
对python中数据集划分函数StratifiedShuffleSplit的使用详解
Dec 11 Python
python random从集合中随机选择元素的方法
Jan 23 Python
Flask使用Pyecharts在单个页面展示多个图表的方法
Aug 05 Python
Python爬取腾讯视频评论的思路详解
Dec 19 Python
python的netCDF4批量处理NC格式文件的操作方法
Mar 21 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
一个显示天气预报的程序
2006/10/09 PHP
php heredoc和phpwind的模板技术使用方法小结
2008/03/28 PHP
PHP 调试工具Debug Tools
2011/04/30 PHP
深入apache配置文件httpd.conf的部分参数说明
2013/06/28 PHP
PHP实时统计中文字数和区别
2019/02/28 PHP
javascript onkeydown,onkeyup,onkeypress,onclick,ondblclick
2009/02/04 Javascript
判断JS对象是否拥有某种属性的两种方式
2013/12/02 Javascript
Node.js编程中客户端Session的使用详解
2015/06/23 Javascript
js+canvas简单绘制圆圈的方法
2016/01/28 Javascript
浅谈JavaScript的全局变量与局部变量
2016/06/10 Javascript
JavaScript定时器制作弹窗小广告
2017/02/05 Javascript
使用JS动态显示文本
2017/09/09 Javascript
微信小程序使用swiper组件实现类3D轮播图
2018/08/29 Javascript
JavaScript链式调用实例浅析
2018/12/19 Javascript
如何优雅地取消 JavaScript 异步任务
2020/03/22 Javascript
Vue 如何使用props、emit实现自定义双向绑定的实现
2020/06/05 Javascript
pymongo实现控制mongodb中数字字段做加法的方法
2015/03/26 Python
python列表的增删改查实例代码
2018/01/30 Python
Python反转序列的方法实例分析
2018/03/21 Python
Django 限制用户访问频率的中间件的实现
2018/08/23 Python
Python3 修改默认环境的方法
2019/02/16 Python
Python类的继承、多态及获取对象信息操作详解
2019/02/28 Python
Python Datetime模块和Calendar模块用法实例分析
2019/04/15 Python
python中必要的名词解释
2019/11/20 Python
使用Rasterio读取栅格数据的实例讲解
2019/11/26 Python
通过python检测字符串的字母
2020/02/18 Python
pycharm中import呈现灰色原因的解决方法
2020/03/04 Python
详解tensorflow2.x版本无法调用gpu的一种解决方法
2020/05/25 Python
大学生简历的个人自我评价
2013/12/04 职场文书
应届毕业生求职信范文
2013/12/18 职场文书
新学期红领巾广播稿
2014/01/14 职场文书
高中班级口号
2014/06/09 职场文书
大学团日活动新闻稿
2014/09/10 职场文书
机关领导查摆四风思想汇报
2014/09/13 职场文书
javascript条件式访问属性和箭头函数介绍
2021/11/17 Javascript
mysql使用instr达到in(字符串)的效果
2022/04/03 MySQL