Como adicionar legenda ao gráfico de dispersão no matplotlib

Você pode usar a seguinte sintaxe para adicionar uma legenda a um gráfico de dispersão no Matplotlib:

 import matplotlib. pyplot as plt
from matplotlib. colors import ListedColormap

#define values, classes, and colors to map 
values = [0, 0, 1, 2, 2, 2]
classes = [' A ', ' B ', ' C ']
colors = ListedColormap([' red ', ' blue ', ' purple '])

#create scatterplot
scatter = plt. scatter (x, y, c=values, cmap=colors)

#add legend
plt. legend ( * scatter.legend_elements ())

Os exemplos a seguir mostram como usar essa sintaxe na prática.

Exemplo 1: Legenda do gráfico de dispersão com valores

O exemplo a seguir mostra como criar um gráfico de dispersão onde a legenda exibe valores:

 import matplotlib. pyplot as plt
from matplotlib. colors import ListedColormap

#define data
x = [3, 4, 4, 6, 8, 9]
y = [12, 14, 17, 16, 11, 13]

#define values, classes, and colors to map 
values = [0, 0, 1, 2, 2, 2]
classes = [' A ', ' B ', ' C ']
colors = ListedColormap([' red ', ' blue ', ' purple '])

#create scatterplot
scatter = plt. scatter (x, y, c=values, cmap=colors)

#add legend with values
plt. legend ( * scatter.legend_elements ()) 

Exemplo 2: Legenda do gráfico de dispersão com classes

O exemplo a seguir mostra como criar um gráfico de dispersão no qual a legenda exibe os nomes das classes:

 import matplotlib. pyplot as plt
from matplotlib. colors import ListedColormap

#define data
x = [3, 4, 4, 6, 8, 9]
y = [12, 14, 17, 16, 11, 13]

#define values, classes, and colors to map 
values = [0, 0, 1, 2, 2, 2]
classes = [' A ', ' B ', ' C ']
colors = ListedColormap([' red ', ' blue ', ' purple '])

#create scatterplot
scatter = plt. scatter (x, y, c=values, cmap=colors)

#add legend with class names
plt. legend (handles=scatter. legend_elements ()[ 0 ], labels=classes)

Observe que esta legenda exibe os nomes das classes que especificamos (A, B, C) em oposição aos valores (0, 1, 2) que especificamos.

Recursos adicionais

Como aumentar o tamanho do gráfico no Matplotlib
Como ajustar a posição do título no Matplotlib
Como definir intervalos de eixos no Matplotlib