La biblioteca Matplotlib es una herramienta muy útil para crear gráficos y visualizaciones en Python. Una de las características más importantes de Matplotlib es la capacidad de crear subplots, es decir, múltiples gráficos en una sola figura. Sin embargo, a veces puede resultar complicado agregar una leyenda a un subplot en Matplotlib. En este artículo, te mostraré cómo hacerlo paso a paso.
1. Importar las bibliotecas necesarias
Antes de comenzar, debemos importar las bibliotecas necesarias. En este caso, necesitaremos importar la biblioteca Matplotlib y NumPy.
«`python
import matplotlib.pyplot as plt
import numpy as np
«`
2. Crear los datos para los subplots
A continuación, debemos crear los datos que utilizaremos para los subplots. En este ejemplo, crearemos dos subplots, uno con una función seno y otro con una función coseno.
«`python
x = np.linspace(0, 2*np.pi, 100)
y1 = np.sin(x)
y2 = np.cos(x)
«`
3. Crear los subplots
Una vez que tenemos los datos, podemos crear los subplots utilizando la función `plt.subplots()`. Esta función nos devuelve una figura y una matriz de ejes, que podemos utilizar para agregar nuestros subplots.
«`python
fig, axs = plt.subplots(2, 1, figsize=(8, 6))
«`
En este ejemplo, creamos una figura con dos subplots en una columna. Especificamos el tamaño de la figura utilizando el parámetro `figsize`.
4. Agregar las leyendas a los subplots
Ahora que tenemos nuestros subplots, podemos agregar las leyendas utilizando la función `axs[i].legend()`, donde `i` es el índice del subplot al que queremos agregar la leyenda.
«`python
axs[0].plot(x, y1, label=’Seno’)
axs[1].plot(x, y2, label=’Coseno’)
axs[0].legend()
axs[1].legend()
«`
En este ejemplo, agregamos la leyenda al primer subplot con la función seno y al segundo subplot con la función coseno. Utilizamos el parámetro `label` para especificar el texto de la leyenda.
5. Mostrar el gráfico
Finalmente, podemos mostrar el gráfico utilizando la función `plt.show()`.
«`python
plt.show()
«`
Esto mostrará la figura con los subplots y las leyendas agregadas.
Agregar una leyenda a un subplot en Matplotlib es bastante sencillo. Solo necesitamos crear los subplots, agregar las leyendas utilizando la función `axs[i].legend()` y mostrar el gráfico utilizando la función `plt.show()`. Con estos pasos, podemos crear visualizaciones más informativas y comprensibles.