Come aggiungere la legenda al grafico a dispersione in matplotlib

È possibile utilizzare la seguente sintassi per aggiungere una legenda a un grafico a dispersione in 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 ())

Gli esempi seguenti mostrano come utilizzare questa sintassi nella pratica.

Esempio 1: legenda del grafico a dispersione con valori

L’esempio seguente mostra come creare un grafico a dispersione in cui la legenda visualizza i valori:

 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 ()) 

Esempio 2: legenda del grafico a dispersione con classi

L’esempio seguente mostra come creare un grafico a dispersione in cui la legenda mostra i nomi delle classi:

 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)

Tieni presente che questa legenda mostra i nomi delle classi che abbiamo specificato (A, B, C) in contrapposizione ai valori (0, 1, 2) che abbiamo specificato.

Risorse addizionali

Come aumentare la dimensione del grafico in Matplotlib
Come regolare la posizione del titolo in Matplotlib
Come impostare gli intervalli degli assi in Matplotlib