Python数据可视化编程通过Matplotlib创建散点图代码示例-CDA数据分析师官网

Python数据可视化编程通过Matplotlib创建散点图代码示例

2018-04-20

Python数据可视化编程通过Matplotlib创建散点图代码示例

Matplotlib简述：

Matplotlib是一个用于创建出高质量图表的桌面绘图包（主要是2D方面）。该项目是由JohnHunter于2002年启动的，其目的是为Python构建一个MATLAB式的绘图接口。如果结合PythonIDE使用比如PyCharm，matplotlib还具有诸如缩放和平移等交互功能。它不仅支持各种操作系统上许多不同的GUI后端，而且还能将图片导出为各种常见的矢量（vector）和光栅（raster）图：PDF、SVG、JPG、PNG、BMP、GIF等。此外，matplotlib还有许多插件工具集，如用于3D图形的mplot3d以及用于地图和投影的basemap。

准备数据：从文本文件中解析数据

本文使用的数据主要包含以下三种特征：每年获得的飞行常客里程数，玩视频游戏所耗时间百分比，每周消费的冰淇淋公升数。其中分类结果作为文件的第四列，并且只有3、2、1三种分类值。

上述特征数据的格式经过file2matrix函数解析处理之后，可输出为矩阵和类标签向量。

将文本记录转换为Numpy的解析程序：

使用file2matrix读取文件数据，必须确保待解析文件存储在当前的工作目录中。导入数据之后，简单检查一下数据格式：

分析数据：使用Matplotlib创建散点图

编辑kNN.py文件，引入matplotlib，调用matplotlib的scatter绘制散点图。

importmatplotlib

importmatplotlib.pyplot as plt

deffile2matrix(filename):

......

returnreturnMat,classLabelVector

datingDataMat,datingLabels=file2matrix('datingTestSet2.txt')

fig=plt.figure()

ax=fig.add_subpot(111)

ax.scatter(datingDataMat[:,1],datingDataMat[:,2])

plt.show()

散点图使用datingDataMat矩阵的第二、第三列数据，分别表示特征值“玩视频游戏所耗时间百分比”和“每周消费的冰淇淋公升数”。

上图由于没有使用样本分类的特征值，很难看到任何有用的数据模式信息。为了更好理解数据信息，Matplotlib库提供的scatter函数支持个性化标记散点图上的点。调用scatter函数使用下列参数：

ax.scatter(datingDataMat[:,1],datingDataMat[:,2],15.0*array(datingLabels),15.0*array(datingLabels))

plt.show()

上图利用datingLabels存储的类标签属性，在散点图上绘制了色彩不等、尺寸不同的点。因而基本上可以从图中看到数据点所属三个样本分类的区域轮廓。为了得到更好的效果，采用datingDataMat矩阵的属性列1和2展示数据，并以红色的'*'表示类标签1、蓝色的'o'表示表示类标签2、绿色的'+'表示类标签3，修改参数如下：

importnumpy as np

......

datingLabels=array(datingLabels)

idx_1=np.where(datingLabels==1)

p1=ax.scatter(datingDataMat[idx_1,0],datingDataMat[idx_1:,1],market='*',color='r',label='1',s=20

idx_2=np.where(datingLabels==2)

p2=ax.scatter(datingDataMat[idx_2,0],datingDataMat[idx_2:,1],market='o',color='b',label='1',s=10

idx_3=np.where(datingLabels==3)

p3=ax.scatter(datingDataMat[idx_3,0],datingDataMat[idx_3:,1],market='+',color='g',label='1',s=30

plt.legend(loc='upper right')

plt.show()

lt.show()

总结

本文简单介绍了Matplotlib，并以实例分析了如何使用Matplotlib库图形化展示数据，最后通过修改matplotlib的scatter函数参数使得散点图的分类区域更加清晰。

CDA数据分析师考试相关入口一览（建议收藏）：

▷ 想报名CDA认证考试，点击>>> “CDA报名” 了解CDA考试详情；

▷ 想加入CDA考试题库，点击>>> “CDA题库” 了解CDA考试详情；

▷ 想学习CDA考试教材，点击>>> “CDA教材” 了解CDA考试详情；

▷ 想查询CDA考试成绩，点击>>> “CDA成绩” 了解CDA考试详情；

▷ 想了解CDA考试含金量，点击>>> “CDA含金量” 了解CDA考试详情；

▷ 想获取CDA考试时间/费用/条件/大纲/通过率，点击 >>>“CDA考试官网” 了解CDA考试详情；

matplotlib 散点图特征 numpy 数据可视化

数据分析咨询请扫描二维码

上一篇如何利用大数据思维来进行用户调研

下一篇中国制造业应如何认识和看待大数据

Python数据可视化编程通过Matplotlib创建散点图代码示例

CDA考试动态

CDA报考指南

热门栏目

最新资讯

数据分析师的工作内容是什么

数据分析师必须掌握的技能有哪些

数据分析入门难吗

数据分析能干啥

数据分析行业的前景

数据分析的常用方法

什么是企业数字化转型

女生做数据分析师最好去哪个行业

大数据技术专业学什么

数据分析师教程《Python数据分析极简入门》第2节 3 ...

什么行业越老越吃香

利用python进行数据分析

大数据与会计专业主要学什么

成为数据分析师需要哪些技能

python可以做数据分析吗

数据分析师的日常工作有哪些？

数据分析五种方法

数字化转型是什么意思

未来发展前景好的十大行业

统计学就业方向及前景