Agregar barra de colores a una gráfica de Matplotlib

Introducción

Matplotlib es una biblioteca de Python utilizada para la visualización de datos. En este laboratorio, aprenderemos cómo agregar una barra de colores a una gráfica en Matplotlib. Las barras de colores son útiles para indicar el rango de valores que representa una mapa de colores.

Consejos sobre la VM

Una vez que se haya iniciado la VM, haga clic en la esquina superior izquierda para cambiar a la pestaña Cuaderno y acceder a Jupyter Notebook para practicar.

A veces, es posible que tenga que esperar unos segundos a que Jupyter Notebook termine de cargarse. La validación de las operaciones no se puede automatizar debido a las limitaciones de Jupyter Notebook.

Si tiene problemas durante el aprendizaje, no dude en preguntar a Labby. Deje comentarios después de la sesión y lo resolveremos rápidamente para usted.

Importar las bibliotecas necesarias

Comenzaremos importando las bibliotecas necesarias. Utilizaremos el módulo pyplot de Matplotlib, que proporciona una interfaz para crear gráficas.

import matplotlib.pyplot as plt

Crear una gráfica

A continuación, crearemos una gráfica utilizando la función imshow de Matplotlib. Esta función muestra una imagen en la gráfica. También crearemos una figura con dos subgráficos.

fig, (ax1, ax2) = plt.subplots(1, 2)
fig.subplots_adjust(wspace=0.5)

im1 = ax1.imshow([[1, 2], [3, 4]])

im2 = ax2.imshow([[1, 2], [3, 4]])

Agregar una barra de colores a la gráfica

Ahora, agregaremos una barra de colores a cada subgráfico utilizando la función make_axes_locatable de Matplotlib. Esta función toma un eje existente, lo agrega a un nuevo AxesDivider y devuelve el AxesDivider. Luego, el método append_axes del AxesDivider se puede utilizar para crear un nuevo eje en un lado dado ("arriba", "derecha", "abajo" o "izquierda") del eje original.

ax1_divider = make_axes_locatable(ax1)
cax1 = ax1_divider.append_axes("right", size="7%", pad="2%")
cb1 = fig.colorbar(im1, cax=cax1)

ax2_divider = make_axes_locatable(ax2)
cax2 = ax2_divider.append_axes("top", size="7%", pad="2%")
cb2 = fig.colorbar(im2, cax=cax2, orientation="horizontal")
cax2.xaxis.set_ticks_position("top")

Mostrar la gráfica

Finalmente, mostraremos la gráfica utilizando la función show de Matplotlib.

plt.show()

Resumen

En este laboratorio, aprendimos cómo agregar una barra de colores a una gráfica en Matplotlib. Utilizamos la función make_axes_locatable para agregar un eje adicional a la gráfica y la función colorbar para crear la barra de colores. También aprendimos cómo cambiar la orientación y la posición de la barra de colores.