wordpress

Matplotlib legend in subplot: Guía para agregar leyenda a subplot

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.

Recomendado:  List All Functions from a Python Module: Available Functions

«`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.

Autor

osceda@hotmail.com

Deja un comentario

Tu dirección de correo electrónico no será publicada. Los campos obligatorios están marcados con *