Создание легенд для графиков Matplotlib

Введение

Matplotlib - это библиотека на Python, которая позволяет пользователям создавать различные виды графиков и диаграмм. В этом руководстве вы узнаете, как создать легенду в графике Matplotlib.

Советы по работе с ВМ

После запуска ВМ щелкните в левом верхнем углу, чтобы переключиться на вкладку Notebook и получить доступ к Jupyter Notebook для практики.

Иногда вам может потребоваться подождать несколько секунд, пока Jupyter Notebook не загрузится полностью. Валидация операций не может быть автоматизирована из-за ограничений в Jupyter Notebook.

Если вы сталкиваетесь с проблемами во время обучения, не стесняйтесь обращаться к Labby. Оставьте отзыв после занятия, и мы оперативно решим проблему для вас.

Импортируем необходимые библиотеки

Прежде чем начать, нам нужно импортировать необходимые библиотеки. В данном случае мы будем использовать библиотеку Matplotlib.

import matplotlib.pyplot as plt

Создаем фигуру и подграфик

Нам нужно создать фигуру и подграфик для построения наших данных. Мы создадим график с двумя подграфиками.

fig = plt.figure()

ax = fig.add_subplot(211)
ax.plot([1, 2, 3], label="test1")
ax.plot([3, 2, 1], label="test2")

ax = fig.add_subplot(223)
ax.plot([1, 2, 3], label="test1")
ax.plot([3, 2, 1], label="test2")

Добавляем легенду к графику

Теперь мы добавим легенду к графику. В Matplotlib есть два способа добавить легенду. Мы будем использовать оба метода в этом примере.

## Method 1: Place a legend above the subplot
ax.legend(bbox_to_anchor=(0., 1.02, 1.,.102), loc='lower left',
           ncols=2, mode="expand", borderaxespad=0.)

## Method 2: Place a legend to the right of the subplot
ax.legend(bbox_to_anchor=(1.05, 1), loc='upper left', borderaxespad=0.)

Отображаем график

Наконец, мы отобразим график.

plt.show()

Резюме

В этом уроке мы узнали, как добавить легенду к графику в Matplotlib. Мы использовали два различных метода для добавления легенды. Первый метод располагал легенду выше подграфика, а второй метод располагал легенду справа от подграфика.