Matplotlib - Gráfico de contorno

Los gráficos de contorno (a veces llamados gráficos de nivel) son una forma de mostrar una superficie tridimensional en un plano bidimensional. Grafica dos variables predictoras XY en el eje y y una variable de respuesta Z como contornos. Estos contornos a veces se denominan segmentos z o valores de respuesta iso.

Un gráfico de contorno es apropiado si desea ver cómo el alor Z cambia en función de dos entradas X e Y, tales que Z = f (X, Y). Una línea de contorno o isolínea de una función de dos variables es una curva a lo largo de la cual la función tiene un valor constante.

Las variables independientes xey suelen estar restringidas a una cuadrícula regular llamada meshgrid. Numpy.meshgrid crea una cuadrícula rectangular a partir de una matriz de valores xy una matriz de valores y.

La API de Matplotlib contiene funciones contour () y contourf () que dibujan curvas de nivel y contornos rellenos, respectivamente. Ambas funciones necesitan tres parámetros x, y y z.

import numpy as np
import matplotlib.pyplot as plt
xlist = np.linspace(-3.0, 3.0, 100)
ylist = np.linspace(-3.0, 3.0, 100)
X, Y = np.meshgrid(xlist, ylist)
Z = np.sqrt(X**2 + Y**2)
fig,ax=plt.subplots(1,1)
cp = ax.contourf(X, Y, Z)
fig.colorbar(cp) # Add a colorbar to a plot
ax.set_title('Filled Contours Plot')
#ax.set_xlabel('x (cm)')
ax.set_ylabel('y (cm)')
plt.show()