¿Cómo agregar una leyenda a un diagrama de dispersión en Matplotlib?

En este artículo, agregaremos una leyenda a las imágenes representadas usando el módulo matplotlib . Usaremos el método matplotlib.pyplot.legend() para describir y etiquetar los elementos del gráfico y distinguir diferentes gráficos del mismo gráfico. 

Sintaxis: matplotlib.pyplot.legend( [“título_1”, “Título_2”], ncol = 1, loc = “superior izquierda”, bbox_to_anchor =(1, 1) )

Parámetros:

• ncol: [toma int, parámetro opcional] el valor predeterminado es 1. Representa el número de columnas en la leyenda.
• loc: [toma una string, parámetro opcional] el valor predeterminado es «mejor», es decir, «arriba a la izquierda». Representa la ubicación de la leyenda. Otras opciones pueden ser: “mejor”, “superior derecha”, “superior izquierda”, “inferior izquierda”, “inferior derecha”, “derecha”, “centro izquierda”, “centro derecha”, “inferior centro”, “superior centro”, “centro”.
• bbox_to_anchor: [toma una lista o tupla de 2 int/float, parámetro opcional]. Representa las coordenadas de la leyenda en el gráfico. Ambas coordenadas xey son obligatorias para dar.

Ejemplo 1: 

# import required modules
import matplotlib.pyplot as plt
 
# adjust coordinates
x =  [1,2,3,4,5]
y1 = [2,4,6,8,10]
y2 = [3,6,9,12,15]
 
# depict illustration
plt.scatter(x, y1)
plt.scatter(x,y2)
 
# apply legend()
plt.legend(["x*2" , "x*3"])
plt.show()

Producción:

Ejemplo 2:

# impoert required modules
import matplotlib.pyplot as plt
 
# adjust coordinates
x =  [1,2,3,4,5]
y1 = [2,4,6,8,10]
y2 = [3,6,9,12,15]
 
# depict illustration
plt.scatter(x, y1)
plt.scatter(x,y2)
 
# apply legend()
plt.legend(["x*2" , "x*3"], ncol = 2 , loc = "lower right")
plt.show()

Producción:

Ejemplo 3:

# import required modules
import matplotlib.pyplot as plt
 
# adjust coordinates
x =  [1,2,3,4,5]
y1 = [2,4,6,8,10]
y2 = [3,6,9,12,15]
 
# depict illustration
plt.scatter(x, y1)
plt.scatter(x,y2)
 
# apply legend()
plt.legend(["x*2" , "x*3"], bbox_to_anchor = (1 , 1))
plt.show()

Producción:

Nota: podemos etiquetar los elementos de la trama en matplotlib.pyplot.scatter() usando el parámetro «etiqueta» . Sin embargo, para mostrarlo se debe escribir  matplotlib.pyplot.legend() .

Ejemplo 4:

# import required modules
import matplotlib.pyplot as plt
import numpy as np
 
# assign coordinates
x = np.arange(1, 6)
y1 = x**2
y2 = x**4
 
# depict illustration
plt.scatter(x, y1, label="x**2")
plt.scatter(x, y2, label="x**4")
 
# apply legend()
plt.legend()
plt.show()

Producción:

Nota: Si matplotlib.pyplot.legend() no está escrito en el código, no se mostrarán las etiquetas.