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

1.matplotlib简介

2.三维图形的绘制过程

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

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

首先导入几个必要的模块

1.  
   import matplotlib .pyplot as plt
2.  
   import 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次方

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

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

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

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

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

设置xyz三轴的标签为XYZ

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

完成绘制

这篇好文章是转载于：学新通技术网