Matplotlib学习日记——3D图

先导包

from mpl_toolkits.mplot3d import Axes3D

再建立一个3D坐标系

fig = plot.figure()
ax = Axes3D(fig)

生成x，y，z坐标列表之后开始画图

ax.plot_surface(x,y,z,rstrid = 1,cstrid  = 1,cmap = plt.get_cmap('rainbow'))

x,y,z分别是坐标列表
rstrid和cstrid分别是行和列的跨度，就是一格一格的跨度
cmap设置颜色渐变，plt.get_cmap()只是其中一种方式，'rainbow’是彩虹色

画出3D图之后对底部加上投影

ax.contourf(x,y,z,zdir = 'z',offset = a,cmap = 'rainbow')  #其实就是一个等高线图

zdir：投影方向，'z’表示从投影图与z轴垂直
offset：投影图的偏移位置
cmpa：颜色渐变