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

Метод 1. Использование параметра labelи функции legend.
Один из самых простых способов добавить легенду к графику – назначить метку каждому графику. элемент с помощью параметра label. После построения графика вызовите функцию legend(), чтобы отобразить легенду.

import matplotlib.pyplot as plt
# Plotting the data
plt.plot(x, y1, label='Line 1')
plt.plot(x, y2, label='Line 2')
# Displaying the legend
plt.legend()
# Displaying the plot
plt.show()

Метод 2: указание местоположения и внешнего вида легенды
Matplotlib предоставляет параметры для настройки местоположения и внешнего вида легенды. Параметр locможно использовать для указания положения легенды (например, «верхний правый», «нижний левый»). Дополнительные параметры, такие как title, fontsizeи frameon, можно использовать для дальнейшей настройки внешнего вида легенды.

# Plotting the data
plt.plot(x, y1, label='Line 1')
plt.plot(x, y2, label='Line 2')
# Displaying the legend at the upper right corner with a custom title and fontsize
plt.legend(loc='upper right', title='Legend', fontsize='small')
# Displaying the plot
plt.show()

Метод 3. Создание легенды с настраиваемыми маркерами
В некоторых случаях может потребоваться создать легенду с настраиваемыми маркерами, например различными фигурами или маркерами. Функция legendпозволяет передавать список пользовательских меток и меток для создания индивидуальной легенды.

import matplotlib.lines as mlines
# Creating custom handles
line1 = mlines.Line2D([], [], color='blue', marker='o', markersize=8, label='Line 1')
line2 = mlines.Line2D([], [], color='green', marker='s', markersize=8, label='Line 2')
# Plotting the data
plt.plot(x, y1)
plt.plot(x, y2)
# Displaying the custom legend
plt.legend(handles=[line1, line2])
# Displaying the plot
plt.show()

Метод 4: добавление легенд к подграфикам
Если у вас есть несколько подграфиков, вы можете добавить отдельные легенды для каждого подграфика. Используйте функцию legendв каждом подграфике, чтобы отобразить соответствующую легенду.

# Creating subplots
fig, axs = plt.subplots(2)
# Plotting the data on each subplot
axs[0].plot(x, y1)
axs[1].plot(x, y2)
# Adding legends to each subplot
axs[0].legend(['Line 1'])
axs[1].legend(['Line 2'])
# Displaying the subplots
plt.show()

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

Следуя этим методам, вы сможете создавать визуально привлекательные графики со значимыми легендами с помощью Matplotlib.