Crear leyendas en diagramas de Matplotlib

Introducción

Matplotlib es una biblioteca de Python que permite a los usuarios crear varios tipos de gráficos y diagramas. Este tutorial lo guiará a través del proceso de creación de una leyenda en un diagrama de 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

Antes de comenzar, necesitamos importar las bibliotecas necesarias. En este caso, usaremos la biblioteca Matplotlib.

import matplotlib.pyplot as plt

Crear una figura y un subdiagrama

Necesitamos crear una figura y un subdiagrama para trazar nuestros datos. Vamos a crear un diagrama con dos subdiagramas.

fig = plt.figure()

ax = fig.add_subplot(211)
ax.plot([1, 2, 3], label="test1")
ax.plot([3, 2, 1], label="test2")

ax = fig.add_subplot(223)
ax.plot([1, 2, 3], label="test1")
ax.plot([3, 2, 1], label="test2")

Agregar una leyenda al diagrama

Ahora agregaremos una leyenda al diagrama. Hay dos maneras de agregar una leyenda en Matplotlib. Usaremos ambos métodos en este ejemplo.

## Método 1: Colocar una leyenda encima del subdiagrama
ax.legend(bbox_to_anchor=(0., 1.02, 1.,.102), loc='lower left',
           ncols=2, mode="expand", borderaxespad=0.)

## Método 2: Colocar una leyenda a la derecha del subdiagrama
ax.legend(bbox_to_anchor=(1.05, 1), loc='upper left', borderaxespad=0.)

Mostrar el diagrama

Finalmente, mostraremos el diagrama.

plt.show()

Resumen

En este tutorial, aprendimos cómo agregar una leyenda a un diagrama de Matplotlib. Usamos dos métodos diferentes para agregar una leyenda. El primer método colocó la leyenda encima del subdiagrama, mientras que el segundo método colocó la leyenda a la derecha del subdiagrama.