如何使用Python和Matplotlib库绘制三维图形?

1.matplotlib简介

2.三维图形的绘制过程

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

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

首先导入几个必要的模块

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

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()

完成绘制