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.