3D图视觉上层次分明，色彩鲜艳，给人以深刻的印象。虽然matplotlib库主要专注于二维图形的绘制，但通过一些扩展包，比如Basemap、GTK工具、Excel工具、AxesGrid和mplot3d等，也能实现3D图形绘制。本节将使用mplot3d扩展包，实现基本的3D绘图功能，绘制简单的三维曲面图形。

【例11.16】绘制简单的三维曲面图。

绘制三维图，会使用如下语句或函数：

①导入mplot3d扩展包：from mpl_toolkits.mplot3d import Axes3D。

②创建Axes3D对象函数：plt.axes（projection="3d"）。

③绘制3D图形：plot_surface（x，y，z，rstride，cstride，cmap）。其中，x，y，z分别表示x轴、y轴和z轴对应数据；rstride和cstride分别表示行步长和列步长；cmap表示颜色图谱映射。

运行结果如图11-20所示。

图11-20　使用matplotlib.pyplot绘制三维图

扫码查看彩图