使用Python--Matplotlib绘制三维图形

1.matplotlib简介

2.三维图形的绘制过程

python画图需要用到matplotlib这个库。具体来说是pylab和pyplot这两个子库。这两个库可以满足基本的画图需求。

如果想画三维图形的话可以直接使用matplotlib.pyplot这个库 通过matplotlib模块的二维绘图功能来绘制三维图形。

首先导入几个必要的模块

import matplotlib .pyplot as pltimport numpy as np

numpy是很重要的模块 它能够生成二维或者三维数组 辅助绘图

接着建立空白图并指定图片大小

fig = plt.figure (figsize= (12,6))

使用.axes方法告诉告诉代码需要返回3d图形

ax = plt.axes(projection="3d")

生成为x y 生成等差数组 计算之后为对z使用.exp方法计算e的-X**2-Y**2次方

X,Y = np.mgrid[-3:3:100j,-3:3:100j]Z = np.exp(-X**2-Y**2)

利用plot_surface函数绘制旋转抛物面 颜色设置为彩色

surf = ax.plot_surface(X,Y,Z,rstride = 1,cstride= 1,cmap = plt.get_cmap('rainbow') )

ax.set_zlim(-0.1,1)ax.set_xlabel("X");ax.set_ylabel("Y");ax.set_zlabel("Z")

限制z轴的坐标（-0.1，1）

设置xyz三轴的标签为XYZ

fig.colorbar(surf,shrink=0.5,aspect=5)plt.show()

完成绘制