как добавить легенду в matplotlib

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

Пример:

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y1 = [1, 2, 3, 4, 5]
y2 = [5, 4, 3, 2, 1]

plt.plot(x, y1, label='График 1')
plt.plot(x, y2, label='График 2')

plt.legend()
plt.show()

В данном примере мы создаем два графика и называем их 'График 1' и 'График 2'. Затем мы вызываем метод legend(), который автоматически добавляет легенду на график.

Если вы хотите расположить легенду в определенном месте на графике, вы можете передать параметр loc в метод legend(). Например:

plt.legend(loc='upper right')

Этот код поместит легенду в верхний правый угол графика. В Matplotlib существует много опций для параметра loc, таких как 'upper left', 'lower right', 'center', и другие.

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