开始我们得用到这几个库numpy,matplolib以及Axes3D,前面两个在我之前的博客中有提到,要是感兴趣的可以看看我前面的博客,我们话不多说直接上代码
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig=plt.figure(figsize=(12,6))#创建一个绘图对象
ax=plt.axes(projection='3d')#用这个绘图对象创建一个Axes对象
#生成代表X轴Y轴数据的列表,并进行网格化
X,Y=np.mgrid[-40:40:0.25,-40:40:0.25]#x轴,y轴的范围是-40到40,
Z=np.sin(np.sqrt(X**2+Y**2))/np.sqrt(X**2+Y**2)#计算函数值
#绘制三维图形
surf=ax.plot_surface(X,Y,Z,
rstride=1,#rstride(row)指定行的跨度
cstride=1,#cstride(column)指定列的跨度
cmap=plt.get_cmap('hot'))#设置颜色映射
ax.set_zlim(-0.1,1)#设置Z轴范围
ax.set_xlabel('X');ax.set_ylabel('Y');ax.set_zlabel('Z')
plt.title('$Z=sin(sqrt(x^2+y^2))/sqrt(x^2+y^2)$')#设置标题
fig.colorbar(surf,shrink=0.5,aspect=5)
plt.show()
效果如下