Введение
В визуализации данных легенда является ключом для интерпретации визуальных элементов графика. Она помогает зрителю понять данные и смысл визуального представления. Matplotlib - это популярная библиотека Python для создания визуализации данных, включая графики с легендами. В этом руководстве мы узнаем, как создать легенду с предварительно определенными метками в Matplotlib.
Советы по работе с ВМ
После запуска ВМ нажмите в верхнем левом углу, чтобы переключиться на вкладку Notebook и получить доступ к Jupyter Notebook для практики.
Иногда вам может потребоваться подождать несколько секунд, пока Jupyter Notebook не загрузится полностью. Валидация операций не может быть автоматизирована из-за ограничений Jupyter Notebook.
Если вы сталкиваетесь с проблемами во время обучения, не стесняйтесь обращаться к Labby. Оставьте отзыв после занятия, и мы оперативно решим проблему для вас.
Импорт необходимых библиотек
Начнем с импорта необходимых библиотек, в которых включаются Matplotlib и NumPy. Мы используем NumPy для генерации некоторых фейковых данных для нашего графика.
import matplotlib.pyplot as plt
import numpy as np
Генерация данных
Далее мы сгенерируем некоторые фейковые данные для использования в нашем графике. Мы создадим два массива, a и b, с использованием функции arange из NumPy. Затем мы вычислим еще два массива, c и d, используя функцию exp для вычисления экспоненты от a, а d будет обратным порядком элементов массива c.
## Make some fake data.
a = b = np.arange(0, 3,.02)
c = np.exp(a)
d = c[::-1]
Создание графика
Теперь мы готовы создать наш график. Мы будем использовать функцию plot из Matplotlib для построения трех линий на одном графике, каждая с предварительно определенной меткой. Мы будем использовать параметр label для присвоения меток каждой линии.
## Create plots with pre-defined labels.
fig, ax = plt.subplots()
ax.plot(a, c, 'k--', label='Model length')
ax.plot(a, d, 'k:', label='Data length')
ax.plot(a, c + d, 'k', label='Total message length')
Добавление легенды
Для добавления легенды к нашему графику мы используем функцию legend из Matplotlib. Мы передаем параметр loc, чтобы указать расположение легенды, и параметр shadow, чтобы добавить эффект тени к легенде. Мы также используем параметр fontsize, чтобы установить размер шрифта легенды.
legend = ax.legend(loc='upper center', shadow=True, fontsize='x-large')
Стилизация легенды
Наконец, мы можем стилизовать легенду, чтобы сделать ее более наглядной. Мы используем функцию get_frame, чтобы получить рамку легенды, а затем функцию set_facecolor, чтобы установить цвет фона рамки.
## Put a nicer background color on the legend.
legend.get_frame().set_facecolor('C0')
Отображение графика
Теперь мы можем отобразить график с использованием функции show из Matplotlib.
plt.show()
Резюме
В этом уроке мы узнали, как создавать легенду с предварительно определенными метками в Matplotlib. Мы использовали функцию plot для построения трех линий на одном графике и параметр label для присвоения меток каждой линии. Затем мы использовали функцию legend для добавления легенды к графику и стилизовали легенду, чтобы сделать ее более наглядной. Следуя этим шагам, вы можете создавать легенды для своих графиков в Matplotlib.