Полезные советы по использованию функции plt.legend

Команда plt.legend() в библиотеке Matplotlib используется для создания легенды на графиках. Легенда — это небольшая область на графике, которая содержит описание каждого элемента, находящегося на графике. Она часто используется для идентификации различных кривых данных или маркеров на графике.

Когда мы построили график с использованием функций в библиотеке Matplotlib, иногда важно иметь возможность объяснить, что означает каждая кривая или каждая точка данных на графике. В этом случае команда plt.legend() становится полезной. Она позволяет нам создать легенду на основе меток, которые мы заранее указали для каждого элемента графика.

Давайте рассмотрим пример, чтобы более подробно разобраться. Вот небольшой код, где мы строим график двух функций - синуса и косинуса:

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 2*np.pi, 100)
sin_y = np.sin(x)
cos_y = np.cos(x)

plt.plot(x, sin_y, label='Синус')
plt.plot(x, cos_y, label='Косинус')

plt.legend()
plt.show()

В этом коде мы сначала импортируем необходимые модули, включая numpy и matplotlib.pyplot. Затем мы создаем массив x, который содержит значения от 0 до 2π с равномерным шагом. Затем мы используем функцию np.sin() и np.cos() из модуля numpy, чтобы получить соответствующие значения синуса и косинуса для каждого значения x.

Затем мы используем функцию plt.plot() для построения линий на графике для полученных значений синуса и косинуса. Важно отметить, что мы указали аргумент label для каждого вызова plt.plot(), чтобы указать метки для каждой линии. Здесь мы использовали метки "Синус" и "Косинус".

После этого, мы вызываем функцию plt.legend(), которая создает легенду на основе указанных меток для каждой линии. И, наконец, мы вызываем plt.show(), чтобы отобразить окно с графиком и его легендой.

При запуске этого кода вы увидите график, содержащий линии для синуса и косинуса, а также легенду, которая объясняет, что означает каждая линия.

Также стоит отметить, что помимо указания меток при построении графика, можно настроить различные параметры легенды, такие как расположение, цвет, размер шрифта, и другие. Для этого можно использовать различные аргументы функции plt.legend(), обеспечивающие гибкость настройки легенды на графике.

Таким образом, команда plt.legend() в библиотеке Matplotlib - это полезный инструмент для добавления легенды на графики, что позволяет лучше понять и интерпретировать данные, представленные на графике.