matplotlib - 3D scatter plot

Matplotlib中也能支持一些基础的3D图表，比如曲面图、散点图和柱状图。这些3D图表需要使用mpl_toolkits模块。

先采样一堆3维的正态分布样本，保证方向上的均匀性；
然后归一化，让每个样本到原点的距离为1，相当于得到了一个均匀分布在球面上的样本；
再接着把每个样本都乘上一个均匀分布随机数的开3次方，这样就得到了在球体内均匀分布的样本；
最后根据判别平面3x+2y-z-1=0对平面两侧样本用不同的形状和颜色画出。

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D  #3D图表需要使用“mpl_toolkits”模块#①构造数据
np.random.seed(42)  #随机数种子
n_samples = 500  # 采样个数为500dim = 3
#生成一组3维正态分布数据，数据方向完全随机
samples = np.random.multivariate_normal(np.zeros(dim), np.eye(dim), n_samples)  #500x3
#通过把每个样本到原点距离和均匀分布吻合得到球体内均匀分布的样本
for i in range(samples.shape[0]):r = np.power(np.random.random(), 1.0/3.0)samples[i] *= r/np.linalg.norm(samples[i])#②将数据分为两组，以3x+2y-z=1作为判别平面
upper_samples = []
lower_samples = []
for x, y, z in samples: if z > 3*x+2*y-1:upper_samples.append((x, y, z))else:lower_samples.append((x, y, z))#③制作散点图
fig = plt.figure('3D scatter plot')
ax = fig.add_subplot(111, projection='3d')  #3d图需要加projection='3d'
uppers = np.array(upper_samples)
lowers = np.array(lower_samples)# 用不同颜色、形状的图标区分两组数据
ax.scatter(uppers[:,0], uppers[:,1], uppers[:,2], c='r', marker='o')  #上半部分为红色圆点
ax.scatter(lowers[:,0], lowers[:,1], lowers[:,2], c='g', marker='^')  #下半部分为绿色三角plt.savefig('3D scatter plot.png')  #保存为图片
plt.show()

参考链接：http://python.jobbole.com/87471/

matplotlib - 3D scatter plot相关推荐

R语言使用car包的scatter3d函数可视化可以交互旋转的3D散点图(Spinning 3D scatter plot by the scatter3d() function of car)
R语言使用car包的scatter3d函数可视化可以交互旋转的3D散点图(Spinning 3D scatter plot produced by the scatter3d() function i ...
cufflinks基于dataframe数据绘制三维散点图（3d scatter plot）
cufflinks基于dataframe数据绘制三维散点图(3d scatter plot) 查看df.iplot对应的各种自定义参数,在jupyter notebook中输入如下信息: df.ipl ...
Python Matplotlib 3D绘图详解（汇总）
Python Matplotlib 3D绘图详解(汇总) 最初开发的 Matplotlib,仅支持绘制 2d 图形,后来随着版本的不断更新, Matplotlib 在二维绘图的基础上,构建了一部分较为 ...
Python使用matplotlib可视化散点图、使用seaborn中的lmplot函数可视化不同分组散点图的最优线性回归拟合曲线（Scatter plot with regression line)
Python使用matplotlib可视化散点图.使用seaborn中的lmplot函数可视化不同分组散点图的最优线性回归拟合曲线(Scatter plot with linear regressio ...
python使用matplotlib可视化3D柱状图（3D bar plot、三维柱状图、包含三个坐标轴x、y、z）、设置zdir参数为y、改变3d图观察的角度
python使用matplotlib可视化3D柱状图(3D bar plot.三维柱状图.包含三个坐标轴x.y.z).设置zdir参数为y.改变3d图观察的角度目录
Python使用matplotlib可视化散点图、可视化两个数值变量之间的关系（Scatter plot）
Python使用matplotlib可视化散点图.可视化两个数值变量之间的关系(Scatter plot) 目录 Python使用matplotlib可视化散点图.可视化两个数值变量之间的关系(Sca
Matplotlib使用scatter函数在Python中绘制气泡图（bubble plot）、通过size参数指定数据点的大小、自定义不同分组的气泡的色彩
Matplotlib使用scatter函数在Python中绘制气泡图(bubble plot).通过size参数指定数据点的大小.自定义不同分组的气泡的色彩目录
Matplotlib使用scatter函数在Python中绘制气泡图（bubble plot）、通过size参数指定数据点的大小
Matplotlib使用scatter函数在Python中绘制气泡图(bubble plot).通过size参数指定数据点的大小目录
python画点的函数_【Python】matplotlib绘图 - scatter、plot函数画离散点（带有边线）...
matplotlib更新之后发现默认的scatter函数绘制的离散点不带有边框.查阅相关文档之后发现需要对其中的参数进行设置. 官方手册:https://matplotlib.org/api/_as_ ...

matplotlib - 3D scatter plot

matplotlib - 3D scatter plot相关推荐

最新文章

热门文章