Як створити легенду вручну в matplotlib (з прикладом)

Ви можете використовувати функції підмодулів matplotlib.lines і matplotlib.patches , щоб створити легенду вручну на діаграмі matplotlib.

У наступному прикладі показано, як це зробити.

Приклад: створити легенду вручну в Matplotlib

Наступний код показує, як створити точкову діаграму в matplotlib із умовною легендою:

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

Щоб створити вручну легенду з власними лініями та квадратами, нам потрібно імпортувати субмодулі matplotlib.lines і matplotlib.patches .

Наступний код показує, як використовувати ці підмодулі для створення легенди вручну:

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

Зверніть увагу, що ця легенда містить мітку вихідних даних, а також мітки та форми елементів, які ми додали вручну.

Щоб змінити мітки або кольори будь-якого з елементів, просто змініть значення аргументів мітки та кольору в попередньому фрагменті коду.

Примітка : зверніться до цього посібника , щоб дізнатися, як змінити положення легенди на графіку.

Додаткові ресурси

У наступних посібниках пояснюється, як виконувати інші типові завдання в Matplotlib:

Як збільшити розмір графіка в Matplotlib
Як налаштувати позицію заголовка в Matplotlib
Як встановити діапазони осей у Matplotlib