Gráfico de contorno en python

¿Cómo haría una cuadrícula de contorno en python utilizando matplotlib.pyplot , donde la cuadrícula es un color donde la variable z está por debajo de cero y otra cuando z es igual o mayor que cero? No estoy muy familiarizado con matplotlib por lo que si alguien me puede dar una forma sencilla de hacerlo, sería genial.

Hasta ahora tengo:

 x= np.arange(0,361) y= np.arange(0,91) X,Y = np.meshgrid(x,y) area = funcarea(L,D,H,W,X,Y) #L,D,H and W are all constants defined elsewhere. plt.figure() plt.contourf(X,Y,area) plt.show() 

Puedes hacer esto usando la palabra clave de levels en contourf.

introduzca la descripción de la imagen aquí

 import numpy as np import matplotlib.pyplot as plt fig, axs = plt.subplots(1,2) x = np.linspace(0, 1, 100) X, Y = np.meshgrid(x, x) Z = np.sin(X)*np.sin(Y) levels = np.linspace(-1, 1, 40) zdata = np.sin(8*X)*np.sin(8*Y) cs = axs[0].contourf(X, Y, zdata, levels=levels) fig.colorbar(cs, ax=axs[0], format="%.2f") cs = axs[1].contourf(X, Y, zdata, levels=[-1,0,1]) fig.colorbar(cs, ax=axs[1]) plt.show() 

Puedes cambiar los colores eligiendo diferentes colores y colores. utilizando vmin, vmax; etc.