Creando barras de color con Matplotlib

Introducción

En este tutorial, aprenderemos a usar la librería matplotlib para crear barras de color para visualizaciones. Las barras de color son una herramienta útil para ayudar a interpretar las visualizaciones, al proporcionar una escala de color que corresponde a los datos que se están graficando. Usaremos matplotlib para crear barras de color para visualizaciones con valores de datos positivos y negativos.

Consejos sobre la VM

Una vez que se haya completado 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 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

Comenzamos importando las bibliotecas necesarias: numpy y matplotlib.pyplot.

import matplotlib.pyplot as plt
import numpy as np

Generar datos

Generamos algunos datos de muestra para graficar, usando la función mgrid de numpy.

## setup some generic data
N = 37
x, y = np.mgrid[:N, :N]
Z = (np.cos(x*0.2) + np.sin(y*0.3))

Crear un gráfico y una barra de color para datos positivos

Creamos un gráfico de los datos positivos y agregamos una barra de color al gráfico usando la función colorbar.

## plot just the positive data and save the
## color "mappable" object returned by ax1.imshow
pos = plt.imshow(Zpos, cmap='Blues', interpolation='none')

## add the colorbar using the figure's method,
## telling which mappable we're talking about and
## which axes object it should be near
plt.colorbar(pos)

Crear un gráfico y una barra de color para datos negativos

Creamos un gráfico de los datos negativos y agregamos una barra de color al gráfico usando la función colorbar. Esta vez, especificamos la ubicación de la barra de color, así como los parámetros de anclaje y encogimiento.

## repeat everything above for the negative data
## you can specify location, anchor and shrink the colorbar
neg = plt.imshow(Zneg, cmap='Reds_r', interpolation='none')
plt.colorbar(neg, location='right', anchor=(0, 0.3), shrink=0.7)

Crear un gráfico con datos positivos y negativos

Creamos un gráfico con datos tanto positivos como negativos, y agregamos una barra de color al gráfico usando la función colorbar. Esta vez, especificamos los valores mínimo y máximo para la barra de color usando los parámetros vmin y vmax.

## Plot both positive and negative values between +/- 1.2
pos_neg_clipped = plt.imshow(Z, cmap='RdBu', vmin=-1.2, vmax=1.2,
                             interpolation='none')

## Add minorticks on the colorbar to make it easy to read the
## values off the colorbar.
cbar = plt.colorbar(pos_neg_clipped, extend='both')
cbar.minorticks_on()

Resumen

En este tutorial, aprendimos cómo usar la librería matplotlib para crear barras de color para visualizaciones. Cubrimos cómo crear barras de color para visualizaciones con valores de datos tanto positivos como negativos. Con estas herramientas, podemos crear visualizaciones más informativas y útiles.