本文实例讲述了Python使用matplotlib绘制三维图形。分享给大家供大家参考，具体如下：
用二维泡泡图表示三维数据
泡泡的坐标2维，泡泡的大小三维，使用到的函数
|
1
2
|
plt.scatter(P[:,0], P[:,1], s=S, lw = 1.5,
edgecolors = C, facecolors='None')
|
其中P[:,0], P[:,1]为泡泡的坐标数据，s为泡泡的大小，lw为泡泡的边线宽度，edgecolors为边线颜色，facecolors为填充颜色
代码及注释
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
# -*- coding:utf-8 -*-
import numpy as np
import matplotlib.pyplot as plt
fig = plt.figure(figsize=(9,6))
# 泡泡的数量
n = 50
size_min = 50
size_max = 50*50
# Ring position,rand可以产生2维数组
P = np.random.rand(n,2)
# Ring colors R,G,B,A
C = np.ones((n,4)) * (0,0,0,1)
# Alpha color channel goes from 0 (transparent) to 1 (opaque)
C[:,3] = np.linspace(0,1,n)
# Ring sizes
S = np.linspace(size_min, size_max, n)
# Scatter plot
plt.scatter(P[:,0], P[:,1], s=S, lw = 1.5,
edgecolors = C, facecolors='None')
#range只能排列出整数，在排列小数时可以用到np.arange
plt.xlim(0,1), plt.xticks(np.arange(0,1.1,0.1))
plt.ylim(0,1), plt.yticks(np.arange(0,1.1,0.1))
plt.show()
|
运行结果：
三维螺旋线图
需要导入三维工具mplot3d
|
1
|
from mpl_toolkits.mplot3d import Axes3D
|
导入了这个工具后才能在add_subplot时添加参数
|
1
|
projection='3d'
|
完整代码及注释
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
# -*- coding:utf-8 -*-
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
import matplotlib.pyplot as plt
from pylab import *
mpl.rcParams['font.sans-serif'] = ['SimHei'] #防止中文乱码
fig = plt.figure(figsize=(9,6))
#添加参数projection
ax = fig.add_subplot(111,projection='3d')
z = np.linspace(0, 6, 1000)
r = 1
x = r * np.sin(np.pi*2*z)
y = r * np.cos(np.pi*2*z)
#c是颜色
ax.plot(x, y, z, label=u'螺旋线', c='r')
ax.legend()
#存成图片，支持png格式
# dpi每英寸长度的点数,画出的三维图是矢量图，可以设置保存的dpi
plt.savefig('3d_fig.png',dpi=200)
plt.show()
|
注：这里针对中文显示问题做了处理，具体可以参考《Python使用matplotlib绘图无法显示中文问题的解决方法》。
参考网站：
https://matplotlib.org/mpl_toolkits/mplot3d/tutorial.html
https://matplotlib.org/gallery.html
希望本文所述对大家Python程序设计有所帮助。
原文链接：https://www.jianshu.com/p/5a30e2896cc0