Matplotlib之3D图像

怎样画出如下所示的3D图像：

代码如下：

import matplotlib.pyplot as plt
from numpy import *
from mpl_toolkits.mplot3d import Axes3D
#设置坐标轴
fig=plt.figure()   #定义一个窗口
ax=Axes3D(fig)     #在窗口添加一个3d坐标轴

#设置X,Y,Z的值
X=arange(-4,4,0.25)  #范围是[-4,4],步长为0.25,生成一个数组
Y=arange(-4,4,0.25)
X,Y=meshgrid(X,Y)    #生成绘制3D图形所需要的网格数据
R=sqrt(X**2+Y**2)
Z=sin(R)

#将图画在3d坐标上
ax.plot_surface(X,Y,Z,rstride=1,cstride=1,cmap=plt.get_cmap('rainbow'),edgecolor='black') #rstride:行跨，cstride:列跨
#将3d图像映射到xoy轴
ax.contourf(X,Y,Z,zdir='Z',offset=-2,cmap='rainbow') #zdir='Z':将z轴压到xoy平面上，offset=-2:将图形映射到z=-2的平面上

ax.set_zlim(-2,2)
plt.show()

注：

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

1. rstride和cstride表示跨度，r:row，c:column。rstride表示行跨，即两条线之间跨了多少行，cstride表示两条线之间垮了多少列。

2. cmap就是图像的颜色属性。

3. edgecolor对应的就是图中一条条线的颜色。

• ax.contourf(X,Y,Z,zdir='Z',offset=-2,cmap='rainbow')

1. 这个函数是画等高线所用到的一个函数，在这个例子中相当于用这个函数将3d图像映射到xoy轴上。

2. zdir='z'，offset=-2：设置一个z=-2的高度，在z轴的方向将这个3d图像压到一个平面上。

• meshgrid(x,y)函数的详细用法：https://blog.csdn.net/beautiful77moon/article/details/97006920