Введение
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. Мы использовали два различных метода для добавления легенды. Первый метод располагал легенду выше подграфика, а второй метод располагал легенду справа от подграфика.