Создание легенды с предварительно определенными метками

PythonPythonBeginner
Практиковаться сейчас

This tutorial is from open-source community. Access the source code

💡 Этот учебник переведен с английского с помощью ИИ. Чтобы просмотреть оригинал, вы можете перейти на английский оригинал

Введение

В визуализации данных легенда является ключом для интерпретации визуальных элементов графика. Она помогает зрителю понять данные и смысл визуального представления. 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.