Python Matplotlib实现三维数据的散点图绘制


Posted in Python onMarch 19, 2019

一、背景

近期项目即将开展,计划第一步就是实现数据的可视化,所以先学习一下数据展示相关Demo。选用Python2.7与Matplotlib来实现,平台采用Pycharm,值得一提的是,Matplotlib的安装前首先要安装Numpy包,但是在完成Numpy的安装之后,楼主不能在PyCharm平台下进行自动安装,或者CMD中使用类似pip install Matplotlib,参考网上解决方案后采用直接去官网下载相应的安装包直接运行安装到相关目录下。在此就不赘述了。

二、 参考

Python语言相对于其他语言对新手较为友好,不用花费太多时间进行语法学习,但是在实际使用的过程中,因为Python中包含有大量的包与资源,在做项目时,对于功能的堆积,实际上Python语言对于新手并不易于理解。相对于Java与C++是需要开发者从底层搭建,可能更易于理解修改(个人意见)。

Python Matplotlib实现三维数据的散点图绘制

三、实现过程

其中就有我们需要参考的部分,也就是mplot3d example code : 2dcollections3d_demo.py。下面贴出其中的代码段。

"""
=======================
Plot 2D data on 3D plot
=======================

Demonstrates using ax.plot's zdir keyword to plot 2D data on
selective axes of a 3D plot.
"""

from mpl_toolkits.mplot3d import Axes3D
import numpy as np
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.gca(projection='3d')

# Plot a sin curve using the x and y axes.
x = np.linspace(0, 1, 100)
y = np.sin(x * 2 * np.pi) / 2 + 0.5
ax.plot(x, y, zs=0, zdir='z', label='curve in (x,y)')

# Plot scatterplot data (20 2D points per colour) on the x and z axes.
colors = ('r', 'g', 'b', 'k')
x = np.random.sample(20*len(colors))
y = np.random.sample(20*len(colors))
c_list = []
for c in colors:
 c_list.append([c]*20)
# By using zdir='y', the y value of these points is fixed to the zs value 0
# and the (x,y) points are plotted on the x and z axes.
ax.scatter(x, y, zs=0, zdir='y', c=c_list, label='points in (x,z)')

# Make legend, set axes limits and labels
ax.legend()
ax.set_xlim(0, 1)
ax.set_ylim(0, 1)
ax.set_zlim(0, 1)
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')

# Customize the view angle so it's easier to see that the scatter points lie
# on the plane y=0
ax.view_init(elev=20., azim=-35)

plt.show()

样例的运行结果大致如下:

Python Matplotlib实现三维数据的散点图绘制

首先样例的数据来自于随机数的产生,但是在我实际使用的过程中,数据是需要预先存储与导入的。因此我添加数据导入部分:

import scipy.io as sio
#get the data form F:\matlab.mat
data = sio.loadmat('F:\matlab.mat')
m = data['data']

值得一提的是这只是我测试的数据,在实际应用过程中,数据的格式是多种多样的,所以需要做数据格式转化的模块。同时采用.mat数据的格式,用户可以用matlab打开,并对数据进行更改之类的操作。采用这种方法导入后,会自动形成数组。

Python Matplotlib实现三维数据的散点图绘制

如上图所示,是数据在matlab中打开的形式,因为我们需要画出三维散点图,会自动产生3×60的数组,每行代表每一维的数据。贴一张做出的Demo的成果图:

Python Matplotlib实现三维数据的散点图绘制

Python Matplotlib实现三维数据的散点图绘制

因为我是用Time变量做为Xlabel,同时模拟数据是等时间间距进行采样的,同时想要在不同的时间点采用不同的颜色。因此需要对ax.scatter(x,y,z,c)中的c变量进行更改,可以用变量代替,这样就可以用个循环结构实现颜色的切换功能。

for a in x:
 if a == 0.1:
  C.append('c')
 elif a == 0.2:
  C.append('r')
 elif a == 0.3:
  C.append('y')
 elif a == 0.4:
  C.append('k')
ax.scatter(x, y, z, c=C)

颜色切换部分代码如下:

import scipy.io as sio
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np

def Singleplot():
 data = sio.loadmat('F:\matlab.mat')
 m = data['data']

 x = m[0]
 y = m[1]
 z = m[2]

 C = []
 ax = plt.subplot(111, projection='3d')

 for a in x:
 if a == 0.1:
  C.append('c')
 elif a == 0.2:
  C.append('r')
 elif a == 0.3:
  C.append('y')
 elif a == 0.4:
  C.append('k')

 ax.scatter(x, y, z, c=C)

 ax.set_xlabel('Time')
 ax.set_ylabel('Frequence')
 ax.set_zlabel('Amplitude')

 plt.show()

singleplot()

需要注意的是Python是属于相对集成度较高的语言,之所以方便使用,是因为存在许多大牛已经完成底层的内容,开发者只需要遵从下载的包中的使用规则,因此过程中对于许多函数不懂的地方,可以用Pycharm的Go to和Declaration功能进入申明区,并且可以从中看到函数的整体介绍,使用语法以及example。因此其中的功能较为有限,如果在短时间内用Python做项目可能对于新手来说,由上到下的形式可能较为简易,但是对于个性化定制功能的话还有待探究。

第一篇博客。记录一下,本周将学习Logistic回归预测,以及部分TensorFlow的原理与Demo,卡尔曼滤波,和一点数据融合算法。该篇属于数据可视化,还会有3D柱状图的绘制与显示将会尽快整理。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python time模块详解(常用函数实例讲解,非常好)
Apr 24 Python
详解Python中内置的NotImplemented类型的用法
Mar 31 Python
Python运算符重载用法实例
May 28 Python
PyCharm+PySpark远程调试的环境配置的方法
Nov 29 Python
python执行CMD指令,并获取返回的方法
Dec 19 Python
Python3批量生成带logo的二维码方法
Jun 24 Python
python如何爬取网站数据并进行数据可视化
Jul 08 Python
关于numpy.where()函数 返回值的解释
Dec 06 Python
pycharm部署、配置anaconda环境的教程
Mar 24 Python
为什么python比较流行
Jun 19 Python
利用python实现汉诺塔游戏
Mar 01 Python
django 认证类配置实现
Nov 11 Python
浅谈python中get pass用法
Mar 19 #Python
使用matplotlib中scatter方法画散点图
Mar 19 #Python
详解django+django-celery+celery的整合实战
Mar 19 #Python
详解Python正则表达式re模块
Mar 19 #Python
python matplotlib画图库学习绘制常用的图
Mar 19 #Python
详解python的四种内置数据结构
Mar 19 #Python
python3使用matplotlib绘制条形图
Mar 25 #Python
You might like
一个显示天气预报的程序
2006/10/09 PHP
一个PHP模板,主要想体现一下思路
2006/12/25 PHP
PHP的Yii框架入门使用教程
2016/02/15 PHP
PHP查询大量数据内存耗尽问题的解决方法
2016/10/28 PHP
php变量与数组相互转换的方法(extract与compact)
2016/12/02 PHP
PHP文字转图片功能原理与实现方法分析
2017/08/31 PHP
jQuery+ajax实现顶一下,踩一下效果
2010/07/17 Javascript
Javascript自定义排序 node运行 实例
2013/06/05 Javascript
js中typeof的用法汇总
2013/12/12 Javascript
node.js中的buffer.copy方法使用说明
2014/12/14 Javascript
JS使用ajax从xml文件动态获取数据显示的方法
2015/03/24 Javascript
js实现点击切换TAB标签实例
2015/08/21 Javascript
微信小程序实现缓存根据不同的id来进行设置和读取缓存
2017/06/12 Javascript
vue使用vue-cli快速创建工程
2017/07/28 Javascript
angularjs中$http异步上传Excel文件方法
2018/02/23 Javascript
为什么说JavaScript预解释是一种毫无节操的机制详析
2018/11/18 Javascript
[48:05]2018DOTA2亚洲邀请赛 3.31 小组赛 B组 VGJ.T vs VP
2018/03/31 DOTA
Python编程中的for循环语句学习教程
2015/10/14 Python
python基础_文件操作实现全文或单行替换的方法
2017/09/04 Python
tensorflow获取变量维度信息
2018/03/10 Python
在dataframe两列日期相减并且得到具体的月数实例
2018/07/03 Python
Django中的Model操作表的实现
2018/07/24 Python
解决Python plt.savefig 保存图片时一片空白的问题
2019/01/10 Python
django 模型中的计算字段实例
2020/05/19 Python
Python参数传递对象的引用原理解析
2020/05/22 Python
整理HTML5中支持的URL编码与字符编码
2016/02/23 HTML / CSS
TecoBuy澳大利亚:在线电子和小工具商店
2020/06/25 全球购物
有abstract方法的类一定要用abstract修饰吗
2016/03/14 面试题
优秀毕业生求职信范文
2014/01/02 职场文书
文明城市标语
2014/06/16 职场文书
工会趣味活动方案
2014/08/18 职场文书
邓小平理论心得体会
2014/09/09 职场文书
2016年少先队活动总结
2016/04/06 职场文书
Java9新特性之Module模块化编程示例演绎
2022/03/16 Java/Android
Java由浅入深通关抽象类与接口(上篇)
2022/04/26 Java/Android
SQL Server中搜索特定的对象
2022/05/25 SQL Server