importmatplotlib.pyplotaspltimportnumpyasnpfrommpl_toolkits.mplot3d.axes3dimportget_test_data# set up a figure twice as wide as it is tallfig=plt.figure(figsize=plt.figaspect(0.5))# =============# First subplot# =============# set up the Axes for the first plotax=fig.add_subplot(1,2,1,projection='3d')# plot a 3D surface like in the example mplot3d/surface3d_demoX=np.arange(-5,5,0.25)Y=np.arange(-5,5,0.25)X,Y=np.meshgrid(X,Y)R=np.sqrt(X**2+Y**2)Z=np.sin(R)surf=ax.plot_surface(X,Y,Z,rstride=1,cstride=1,cmap="coolwarm",linewidth=0,antialiased=False)ax.set_zlim(-1.01,1.01)fig.colorbar(surf,shrink=0.5,aspect=10)# ==============# Second subplot# ==============# set up the Axes for the second plotax=fig.add_subplot(1,2,2,projection='3d')# plot a 3D wireframe like in the example mplot3d/wire3d_demoX,Y,Z=get_test_data(0.05)ax.plot_wireframe(X,Y,Z,rstride=10,cstride=10)plt.show()

DownloadJupyternotebook:subplot3d.ipynb

DownloadPythonsourcecode:subplot3d.py

Downloadzipped:subplot3d.zip