Hoe een legenda aan een spreidingsdiagram in matplotlib toe te voegen

U kunt de volgende syntaxis gebruiken om een legenda toe te voegen aan een spreidingsdiagram 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 ())

De volgende voorbeelden laten zien hoe u deze syntaxis in de praktijk kunt gebruiken.

Voorbeeld 1: Scatterplot-legenda met waarden

In het volgende voorbeeld ziet u hoe u een spreidingsdiagram maakt waarin de legenda waarden weergeeft:

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

Voorbeeld 2: Scatterplot-legenda met klassen

Het volgende voorbeeld laat zien hoe u een spreidingsdiagram maakt waarin de legenda de klassennamen weergeeft:

 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)

Merk op dat deze legenda de klassenamen weergeeft die we hebben opgegeven (A, B, C) in tegenstelling tot de waarden (0, 1, 2) die we hebben opgegeven.

Aanvullende bronnen

Hoe de plotgrootte in Matplotlib te vergroten
Hoe de titelpositie in Matplotlib aan te passen
Hoe asbereiken in Matplotlib in te stellen