Note
Go to the endto download the full example code.
2D and 3D Axes in same figure#
This example shows a how to plot a 2D and a 3D plot on the same figure.
importmatplotlib.pyplotaspltimportnumpyasnpdeff(t):returnnp.cos(2*np.pi*t)*np.exp(-t)# Set up a figure twice as tall as it is widefig=plt.figure(figsize=plt.figaspect(2.))fig.suptitle('A tale of 2 subplots')# First subplotax=fig.add_subplot(2,1,1)t1=np.arange(0.0,5.0,0.1)t2=np.arange(0.0,5.0,0.02)t3=np.arange(0.0,2.0,0.01)ax.plot(t1,f(t1),'bo',t2,f(t2),'k--',markerfacecolor='green')ax.grid(True)ax.set_ylabel('Damped oscillation')# Second subplotax=fig.add_subplot(2,1,2,projection='3d')X=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,linewidth=0,antialiased=False)ax.set_zlim(-1,1)plt.show()

Tags:plot-type: 3Dcomponent: subplotlevel: beginner