Principiante de Python que pide ayuda, use matplotlib para dibujar mapas de contorno, z es una matriz 2D

Tome el dibujo de la línea de contorno de ?z=x^2 y^2? como ejemplo para presentar brevemente el método de dibujar líneas de contorno con matplotlib.

Primero mire ?z = Imagen tridimensional de x^2 y^2: import?numpy?as?npfrom?matplotlib?import?cmimport?matplotlib.pyplot?as?pltfrom?mpl_toolkits.mplot3d?import?Axes3D

delta ?=? 0.2x?=?np.arange(-3,?3,?delta)

y?=?np.arange(-3,?3,?delta)

X ,?Y?=?np.meshgrid(x,?y)

Z?=?X**2? ?Y**2x=X.flatten()

y =Y.flatten()

z=Z.flatten()

fig?=?plt.figure()

ax?=? fig.gca (projection='3d')

ax.plot_trisurf(x,?y,?z,?cmap=cm.jet,?linewidth=0.01)

plt. show()

A continuación se dibujan las líneas de contorno de esta figura: plt.contour(X,?Y,?Z)?#Tenga en cuenta que está en mayúscula X?, Y, Z

Si solo ¿Quieres dibujar estas tres líneas de contorno?z=1,z=2,z=3?: plt.contour(X,?Y,?Z,)