PLT Legend - все о легендах графиков в Python
plt.legend - это функция, которая добавляет легенду к графику. Легенда позволяет объяснить, что означают линии и точки на графике.
Пример использования plt.legend:
import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0, 10, num=100)
y1 = np.sin(x)
y2 = np.cos(x)
plt.plot(x, y1, label='sin(x)')
plt.plot(x, y2, label='cos(x)')
plt.legend()
plt.show()
В этом примере мы создаем два графика: sin(x) и cos(x), используя numpy для создания массивов данных. Затем мы используем метод plt.plot для построения графика каждой из функций. Флаг label задает ярлык для каждой линии.
Затем мы используем plt.legend() без аргументов, чтобы вывести легенду на графике. Matplotlib автоматически создаст список ярлыков на основе значений label.
Другой пример с plt.legend:
import matplotlib.pyplot as plt
x = [1, 2, 3, 4, 5]
y = [10, 20, 15, 25, 30]
plt.plot(x, y, 'ro', label='Data')
plt.ylabel('Y axis')
plt.xlabel('X axis')
plt.legend(loc='upper left')
plt.show()
Здесь мы отображаем не функцию, а просто набор данных с помощью метода plt.plot с флагом 'ro', который указывает на красные кружки. У нас также есть оси X и Y, а также ярлык 'Data', указывающий на наш набор данных.
Затем мы используем параметр loc, чтобы указать местоположение легенды. В данном случае мы используем loc='upper left', чтобы поместить легенду в левый верхний угол графика.
В обоих приведенных выше примерах мы использовали plt.legend(), чтобы добавить легенду к графику. Эта функция может принимать различные аргументы и параметры, такие как:
- labels: список ярлыков для отображения на легенде
- loc: указывает местоположение легенды
- fontsize: задает размер шрифта ярлыков на легенде
- title: Задает заголовок легенды
Таким образом, plt.legend позволяет создавать понятные и информативные графики, которые помогают проиллюстрировать данные и результаты исследований.