Python使用matplotlib绘制三维图形示例
本文实例讲述了Python使用matplotlib绘制三维图形。分享给大家供大家参考,具体如下:
用二维泡泡图表示三维数据
泡泡的坐标2维,泡泡的大小三维,使用到的函数
plt.scatter(P[:,0], P[:,1], s=S, lw = 1.5, edgecolors = C, facecolors='None')
其中P[:,0], P[:,1]为泡泡的坐标数据,s为泡泡的大小,lw为泡泡的边线宽度,edgecolors为边线颜色,facecolors为填充颜色
代码及注释
# -*- 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
from mpl_toolkits.mplot3d import Axes3D
导入了这个工具后才能在add_subplot时添加参数
projection='3d'
完整代码及注释
# -*- 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相关内容感兴趣的读者可查看本站专题:《Python数学运算技巧总结》、《Python数据结构与算法教程》、《Python函数使用技巧总结》、《Python字符串操作技巧汇总》、《Python入门与进阶经典教程》及《Python文件与目录操作技巧汇总》
希望本文所述对大家Python程序设计有所帮助。