Cómo crear una gráfica de contorno en Matplotlib

Un gráfico de contorno es un tipo de gráfico que nos permite visualizar datos tridimensionales en dos dimensiones mediante el uso de contornos.

Puede crear una gráfica de contorno en Matplotlib usando las siguientes dos funciones:

Los siguientes ejemplos muestran cómo utilizar estas dos funciones en la práctica.

Ejemplo 1: Gráfico de contorno en Matplotlib

Supongamos que tenemos los siguientes datos en Python:

importar numpy como np

x = np.linspace (0, 5, 50)
y = np.linspace (0, 5, 40)

X, Y = np.meshgrid (x, y)
Z = np.sin (X * 2 + Y) * 3 + np.cos (Y + 5)

Podemos usar el siguiente código para crear una gráfica de contorno para los datos:

importar matplotlib.pyplot como plt

plt. contorno (X, Y, Z, colores = ' negro ')

Mapa de contorno en matplotlib

Cuando se usa un solo color para el gráfico, las líneas discontinuas representan valores negativos y las líneas continuas representan valores positivos.

Una alternativa es especificar un mapa de colores usando el argumento cmap . También podemos especificar más líneas para usar en la trama con el argumento de niveles :

plt. contorno (X, Y, Z, niveles = 30 , cmap = ' Rojos ')

Mapa de contorno de matplotlib con cmap

Elegimos usar el cmap ‘Reds’ pero puede encontrar una lista completa de opciones de mapas de colores en la página de documentación de Matplotlib .

Ejemplo 2: Gráfico de contorno relleno en Matplotlib

Un gráfico de contorno relleno es similar a un gráfico de contorno excepto que los espacios entre las líneas están rellenos.

El siguiente código muestra cómo usar la función contourf () para crear un gráfico de contorno relleno para los mismos datos que usamos en el ejemplo anterior:

plt. contourf (X, Y, Z, cmap = ' Rojos ')

Gráfico de contorno relleno en Matplotlib

También podemos usar la función colorbar () para agregar una barra de color etiquetada junto al gráfico:

plt. contourf (X, Y, Z, cmap = ' Rojos ')
plt. barra de colores ()

Mapa de contorno con barra de colores en Matplotlib

Puede encontrar más tutoriales de Matplotlib aquí .

  • https://r-project.org
  • https://www.python.org/
  • https://www.stata.com/

Deja un comentario

La distancia de Mahalanobis es la distancia entre dos puntos en un espacio multivariado. A menudo se usa para encontrar…
statologos comunidad-2

Compartimos información EXCLUSIVA y GRATUITA solo para suscriptores (cursos privados, programas, consejos y mucho más)

You have Successfully Subscribed!