Cara membuat legenda manual di matplotlib (dengan contoh)

Anda dapat menggunakan fungsi submodul matplotlib.lines dan matplotlib.patches untuk membuat legenda manual di plot matplotlib.

Contoh berikut menunjukkan cara melakukan hal ini.

Contoh: membuat legenda manual di Matplotlib

Kode berikut menunjukkan cara membuat scatterplot di matplotlib dengan legenda default:

 import matplotlib. pyplot as plt

#define data to plot
x = [1, 2, 3, 4, 5, 6, 7]
y = [2, 3, 5, 8, 12, 18, 27]

#create scatter plot of x vs. y
plt. scatter (x, y, label=' Original Data ', color=' steelblue ')

#add legend
plt. legend ()

#displayplot
plt. show () 

Untuk membuat legenda manual dengan garis dan kotak khusus, kita perlu mengimpor submodul matplotlib.lines dan matplotlib.patches .

Kode berikut menunjukkan cara menggunakan submodul ini untuk membuat legenda manual:

 import matplotlib. pyplot as plt
from matplotlib. lines import Line2D
import matplotlib. patches as mpatches

#define data to plot
x = [1, 2, 3, 4, 5, 6, 7]
y = [2, 3, 5, 8, 12, 18, 27]

#create scatter plot of x vs. y
plt. scatter (x, y, label=' Original Data ', color=' steelblue ')

#define handles and labels that will get added to legend
handles, labels = plt. gca (). get_legend_handles_labels ()

#define patches and lines to add to legend
patch1 = mpatches. Patch (color=' orange ', label=' First Manual Patch ')
patch2 = mpatches. Patch (color=' orange ', label=' First Manual Patch ')   
line1 = Line2D([0], [0], label=' First Manual Line ', color=' purple ')
line2 = Line2D([0], [0], label=' Second Manual Line ', color=' red ')

#add handles
handles. extend ([patch1, line1, line2])

#add legend
plt. legend (handles=handles)

#displayplot
plt. show () 

Perhatikan bahwa legenda ini mencakup label data asli, tetapi juga label dan bentuk elemen yang kami tambahkan secara manual.

Untuk mengubah label atau warna elemen mana pun, cukup ubah nilai argumen label dan warna pada potongan kode sebelumnya.

Catatan : Lihat tutorial ini untuk mempelajari cara mengubah posisi legenda dalam plot.

Sumber daya tambahan

Tutorial berikut menjelaskan cara melakukan tugas umum lainnya di Matplotlib:

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