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入门篇之字典
Oct 17 Python
Python抓取手机号归属地信息示例代码
Nov 28 Python
浅谈numpy中linspace的用法 (等差数列创建函数)
Jun 07 Python
致Python初学者 Anaconda入门使用指南完整版
Apr 05 Python
Python利用matplotlib.pyplot绘图时如何设置坐标轴刻度
Apr 09 Python
在Python 不同级目录之间模块的调用方法
Jan 19 Python
python安装requests库的实例代码
Jun 25 Python
Python数据可视化:泊松分布详解
Dec 07 Python
python读取ini配置的类封装代码实例
Jan 08 Python
PyTorch实现ResNet50、ResNet101和ResNet152示例
Jan 14 Python
全网首秀之Pycharm十大实用技巧(推荐)
Apr 27 Python
python批量生成条形码的示例
Oct 10 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 array_intersect()函数使用代码
2009/01/14 PHP
PHP GD库生成图像的几个函数总结
2014/11/19 PHP
PHP6连接SQLServer2005的三部曲
2016/04/15 PHP
JavaScript语言中的Literal Syntax特性分析
2007/03/08 Javascript
juqery 学习之三 选择器 可见性 元素属性
2010/11/25 Javascript
jQuery 选择器、DOM操作、事件、动画
2010/11/25 Javascript
在多个页面使用同一个HTML片段《续》
2011/03/04 Javascript
js中typeof的用法汇总
2013/12/12 Javascript
js清空表单数据的两种方式(遍历+reset)
2014/07/18 Javascript
jQuery实现设置、移除文本框默认值功能
2015/01/13 Javascript
Javascript数组操作函数总结
2015/02/05 Javascript
jquery判断复选框选中状态以及区分attr和prop
2015/12/18 Javascript
jquery及js实现动态加载js文件的方法
2016/01/21 Javascript
详解jQuery中的DOM操作
2016/12/23 Javascript
javascript工厂模式和构造函数模式创建对象方法解析
2016/12/30 Javascript
bootstrap模态框关闭后清除模态框的数据方法
2018/08/10 Javascript
vue中eslintrc.js配置最详细介绍
2018/12/21 Javascript
uni-app如何页面传参数的几种方法总结
2020/04/28 Javascript
从Node.js事件触发器到Vue自定义事件的深入讲解
2020/06/26 Javascript
使用python实现baidu hi自动登录的代码
2013/02/10 Python
Python变量和数据类型详解
2017/02/15 Python
python正则实现提取电话功能
2018/02/24 Python
python 编码规范整理
2018/05/05 Python
Python连接Mssql基础教程之Python库pymssql
2018/09/16 Python
Python使用QQ邮箱发送邮件实例与QQ邮箱设置详解
2020/02/18 Python
Helly Hansen工作服美国官方网上商店:为最恶劣的环境
2019/09/04 全球购物
生物医学工程专业学生求职信范文分享
2013/12/14 职场文书
心理健康心得体会
2014/01/02 职场文书
《小白兔和小灰兔》教学反思
2014/02/18 职场文书
挂职自我鉴定
2014/02/26 职场文书
农村葬礼主持词
2014/03/31 职场文书
教学改革实施方案
2014/03/31 职场文书
《分一分》教学反思
2014/04/13 职场文书
文明倡议书范文
2014/04/15 职场文书
物业管理委托协议(2篇)
2014/09/23 职场文书
领导班子四风对照检查材料范文
2014/09/27 职场文书