Bagaimana cara menambahkan legenda ke scatterplot di matplotlib

Anda dapat menggunakan sintaks berikut untuk menambahkan legenda ke sebar di 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 ())

Contoh berikut menunjukkan cara menggunakan sintaksis ini dalam praktiknya.

Contoh 1: Legenda sebar dengan nilai

Contoh berikut menunjukkan cara membuat plot sebar di mana legenda menampilkan nilai:

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

Contoh 2: Legenda sebar dengan kelas

Contoh berikut menunjukkan cara membuat plot sebar di mana legenda menampilkan nama kelas:

 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)

Perhatikan bahwa legenda ini menampilkan nama kelas yang kami tentukan (A, B, C) sebagai lawan dari nilai (0, 1, 2) yang kami tentukan.

Sumber daya tambahan

Cara menambah ukuran plot di Matplotlib
Cara menyesuaikan posisi judul di Matplotlib
Cara mengatur rentang sumbu di Matplotlib