Creando una leyenda con etiquetas predefinidas

Introducción

En la visualización de datos, una leyenda es la clave para interpretar los elementos visuales de una trama. Ayuda al espectador a entender los datos y el significado de la representación visual. Matplotlib es una popular biblioteca de Python para crear visualizaciones de datos, incluyendo tramas con leyendas. En este tutorial, aprenderemos cómo crear una leyenda con etiquetas predefinidas 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 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.

Importando las bibliotecas necesarias

Comenzaremos importando las bibliotecas necesarias, que incluyen Matplotlib y NumPy. Usamos NumPy para generar algunos datos falsos para nuestra trama.

import matplotlib.pyplot as plt
import numpy as np

Generando los datos

A continuación, generaremos algunos datos falsos para usar en nuestra trama. Crearemos dos arrays, a y b, usando la función arange de NumPy. Luego calcularemos otros dos arrays, c y d, usando la función exp para calcular el exponencial de a y d como el inverso de c.

## Make some fake data.
a = b = np.arange(0, 3,.02)
c = np.exp(a)
d = c[::-1]

Creando la trama

Ahora estamos listos para crear nuestra trama. Usaremos la función plot de Matplotlib para trazar tres líneas en la misma gráfica, cada una con una etiqueta predefinida. Usaremos el parámetro label para asignar las etiquetas a cada línea.

## Create plots with pre-defined labels.
fig, ax = plt.subplots()
ax.plot(a, c, 'k--', label='Model length')
ax.plot(a, d, 'k:', label='Data length')
ax.plot(a, c + d, 'k', label='Total message length')

Agregando la leyenda

Para agregar la leyenda a nuestra trama, usamos la función legend de Matplotlib. Pasamos el parámetro loc para especificar la ubicación de la leyenda y el parámetro shadow para agregar un efecto de sombra a la leyenda. También usamos el parámetro fontsize para establecer el tamaño de fuente de la leyenda.

legend = ax.legend(loc='upper center', shadow=True, fontsize='x-large')

Dar estilo a la leyenda

Finalmente, podemos dar estilo a la leyenda para que sea más atractiva visualmente. Usamos la función get_frame para obtener el marco de la leyenda y luego la función set_facecolor para establecer el color de fondo del marco.

## Put a nicer background color on the legend.
legend.get_frame().set_facecolor('C0')

Mostrando la trama

Ahora podemos mostrar la trama usando la función show de Matplotlib.

plt.show()

Resumen

En este tutorial, aprendimos cómo crear una leyenda con etiquetas predefinidas en Matplotlib. Usamos la función plot para trazar tres líneas en la misma gráfica y usamos el parámetro label para asignar etiquetas a cada línea. Luego usamos la función legend para agregar la leyenda a la trama y le dimos estilo para que fuera más atractiva visualmente. Siguiendo estos pasos, puedes crear leyendas para tus propias tramasy en Matplotlib.