Matplotlib.pyplot.legend() en Python

Matplotlib es uno de los paquetes de Python más populares utilizados para la visualización de datos. Es una biblioteca multiplataforma para hacer gráficos 2D a partir de datos en arrays. Pyplot es una colección de funciones de estilo de comando que hacen que matplotlib funcione como MATLAB. Cada función de pyplot realiza algún cambio en una figura: por ejemplo, crea una figura, crea un área de trazado en una figura, traza algunas líneas en un área de trazado, decora el trazado con etiquetas, etc.

Matplotlib.pyplot.legend()

Una leyenda es un área que describe los elementos del gráfico. En la biblioteca matplotlib, hay una función llamada legend() que se usa para colocar una leyenda en los ejes.

El atributo Loc in legend()se utiliza para especificar la ubicación de la leyenda. El valor predeterminado de loc es loc=”mejor” (parte superior izquierda). Las strings ‘superior izquierda’, ‘superior derecha’, ‘inferior izquierda’, ‘inferior derecha’ colocan la leyenda en la esquina correspondiente de los ejes/figura.

El atributo bbox_to_anchor=(x, y) de la función legend() se usa para especificar las coordenadas de la leyenda, y el atributo ncol representa el número de columnas que tiene la leyenda. Su valor predeterminado es 1.

Sintaxis:

matplotlib.pyplot.legend([“azul”, “verde”], bbox_to_anchor=(0.75, 1.15), ncol=2)

Los siguientes son algunos atributos más de la función legend():

• sombra : [Ninguno o bool] Si dibujar una sombra detrás de la leyenda. Su valor predeterminado es Ninguno.
• markerscale : [Ninguno o int o float] El tamaño relativo de los marcadores de leyenda en comparación con los dibujados originalmente. El valor predeterminado es Ninguno.
• numpoints : [Ninguno o int] El número de puntos de marcador en la leyenda al crear una entrada de leyenda para una Línea2D (línea). El valor predeterminado es Ninguno.
• fontsize : El tamaño de fuente de la leyenda. Si el valor es numérico, el tamaño será el tamaño de fuente absoluto en puntos.
• facecolor : [Ninguno o “heredar” o color] El color de fondo de la leyenda.
• edgecolor : [Ninguno o “heredar” o color] El color del borde del parche de fondo de la leyenda.

Maneras de usar la función legend() en Python –

Ejemplo 1:

import numpy as np
import matplotlib.pyplot as plt
  
# X-axis values
x = [1, 2, 3, 4, 5]
  
# Y-axis values 
y = [1, 4, 9, 16, 25]
  
# Function to plot  
plt.plot(x, y)
  
# Function add a legend  
plt.legend(['single element'])
  
# function to show the plot
plt.show()

Producción :

Ejemplo 2:

# importing modules
import numpy as np
import matplotlib.pyplot as plt
  
# Y-axis values
y1 = [2, 3, 4.5]
  
# Y-axis values 
y2 = [1, 1.5, 5]
  
# Function to plot  
plt.plot(y1)
plt.plot(y2)
  
# Function add a legend  
plt.legend(["blue", "green"], loc ="lower right")
  
# function to show the plot
plt.show()

Producción :

Ejemplo 3:

import numpy as np
import matplotlib.pyplot as plt
  
# X-axis values
x = np.arange(5)
  
# Y-axis values
y1 = [1, 2, 3, 4, 5]
  
# Y-axis values 
y2 = [1, 4, 9, 16, 25]
  
# Function to plot  
plt.plot(x, y1, label ='Numbers')
plt.plot(x, y2, label ='Square of numbers')
  
# Function add a legend  
plt.legend()
  
# function to show the plot
plt.show()

Producción :

Ejemplo 4:

import numpy as np
import matplotlib.pyplot as plt
  
x = np.linspace(0, 10, 1000)
fig, ax = plt.subplots()
  
ax.plot(x, np.sin(x), '--b', label ='Sine')
ax.plot(x, np.cos(x), c ='r', label ='Cosine')
ax.axis('equal')
  
leg = ax.legend(loc ="lower left");

Producción:

Ejemplo 5:

# importing modules
import numpy as np
import matplotlib.pyplot as plt
   
# X-axis values
x = [0, 1, 2, 3, 4, 5, 6, 7, 8]
   
# Y-axis values
y1 = [0, 3, 6, 9, 12, 15, 18, 21, 24]
# Y-axis values 
y2 = [0, 1, 2, 3, 4, 5, 6, 7, 8]
   
# Function to plot  
plt.plot(y1, label ="y = x")
plt.plot(y2, label ="y = 3x")
   
# Function add a legend  
plt.legend(bbox_to_anchor =(0.75, 1.15), ncol = 2)
   
# function to show the plot
plt.show()

Producción: