Crear etiquetas personalizadas de valores de graduación en la barra de color de Matplotlib

Introducción

En la visualización de datos, las barras de color se utilizan para representar el rango de valores de un conjunto de datos utilizando el color. Matplotlib es una biblioteca de Python para crear una variedad de visualizaciones, incluyendo barras de color. En este laboratorio, aprenderemos cómo personalizar las etiquetas de los valores de graduación en una barra de color en Matplotlib.

Consejos sobre la VM

Una vez finalizada la inicialización de 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 sus comentarios después de la sesión y lo resolveremos rápidamente para usted.

Importar las bibliotecas necesarias y fijar el estado aleatorio

Primero, necesitamos importar las bibliotecas necesarias y fijar el estado aleatorio para la reproducibilidad. Usaremos numpy para generar algunos datos aleatorios, matplotlib.pyplot para crear visualizaciones y cm de matplotlib para definir las mapas de color.

import matplotlib.pyplot as plt
import numpy as np
from numpy.random import randn

from matplotlib import cm

## Fixing random state for reproducibility
np.random.seed(19680801)

Crear un gráfico con una barra de color vertical

Comenzaremos creando un gráfico con una barra de color vertical. Generaremos algunos datos aleatorios utilizando randn de numpy y recortaremos los valores al rango de -1 a 1. Luego crearemos un objeto AxesImage utilizando imshow y la paleta de colores coolwarm. Finalmente, agregaremos un título al gráfico.

## Make plot with vertical (default) colorbar
fig, ax = plt.subplots()

data = np.clip(randn(250, 250), -1, 1)

cax = ax.imshow(data, cmap=cm.coolwarm)
ax.set_title('Gaussian noise with vertical colorbar')

Personalizar las etiquetas de los valores de graduación en la barra de color vertical

A continuación, personalizaremos las etiquetas de los valores de graduación en la barra de color vertical. Crearemos una barra de color utilizando colorbar y especificaremos las ubicaciones de los valores de graduación utilizando el parámetro ticks. Luego estableceremos las etiquetas de los valores de graduación utilizando set_yticklabels en el atributo ax del objeto de barra de color.

## Add colorbar, make sure to specify tick locations to match desired ticklabels
cbar = fig.colorbar(cax, ticks=[-1, 0, 1])
cbar.ax.set_yticklabels(['< -1', '0', '> 1'])  ## vertically oriented colorbar

Crear un gráfico con una barra de color horizontal

Ahora crearemos un gráfico con una barra de color horizontal. Seguiremos los mismos pasos que en el Paso 2, pero esta vez usaremos la paleta de colores afmhot y estableceremos la orientación de la barra de color en horizontal.

## Make plot with horizontal colorbar
fig, ax = plt.subplots()

data = np.clip(randn(250, 250), -1, 1)

cax = ax.imshow(data, cmap=cm.afmhot)
ax.set_title('Gaussian noise with horizontal colorbar')

cbar = fig.colorbar(cax, ticks=[-1, 0, 1], orientation='horizontal')
cbar.ax.set_xticklabels(['Low', 'Medium', 'High'])  ## horizontal colorbar

Mostrar el gráfico

Finalmente, mostraremos el gráfico utilizando plt.show().

plt.show()

Resumen

En este laboratorio, aprendimos cómo personalizar las etiquetas de los valores de graduación en una barra de color en Matplotlib. Primero creamos un gráfico con una barra de color vertical y personalizamos las etiquetas de los valores de graduación utilizando set_yticklabels. Luego creamos un gráfico con una barra de color horizontal y personalizamos las etiquetas de los valores de graduación utilizando set_xticklabels. Personalizar las etiquetas de los valores de graduación en una barra de color puede ayudar a hacer que tus visualizaciones sean más informativas y fáciles de leer.