使用matplotlab绘制3D视图

使用matplotlab绘制3D图像，首先需要引入一个3D模块mpl_toolkits.mplot3d中的Axes3D,绘制图像fig=plt.figure()
,绘制3D图像ax=Axes3D()，直接上代码：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig=plt.figure()
ax=Axes3D(fig)
x=np.arange(-4,4,0.25)
y=np.arange(-4,4,0.25)
x,y=np.meshgrid(x,y)#生成网格数据
L=np.sqrt(x**2+y**2)#计算每个点对的长度
z=np.square(L)#计算z轴的高度
ax.set_xlabel('X',fontsize=18)
ax.set_ylabel('y',fontsize=18)
ax.set_zlabel('z',fontsize=18)
ax.plot_surface(x,y,z,rstride=1,cstride=1,cmap='rainbow')
plt.savefig("3D_square.jpg")
plt.show()

绘制结果如图