📊 Как создать легенду в matplotlib? Инструкция для начинающих

Чтобы создать легенду в matplotlib, вы можете использовать метод legend() для текущей активной фигуры или оси.

import matplotlib.pyplot as plt

# Создание данных
x = [1, 2, 3, 4, 5]
y = [10, 5, 7, 9, 3]

# Построение графика
plt.plot(x, y, label='Данные')

# Добавление легенды
plt.legend()

# Отображение графика
plt.show()

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

Помните, что вы также можете настраивать легенду, добавляя параметры, такие как положение, цвет и шрифт.

Как сделать легенду в matplotlib

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

Создание простой легенды

Чтобы создать простую легенду, необходимо использовать функцию legend() из модуля matplotlib.pyplot. Эта функция принимает несколько параметров, включая список меток и положение легенды на графике.

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 2*np.pi, 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()

В приведенном выше примере мы создаем массив x с использованием функции linspace() из модуля numpy. Затем мы вычисляем значения синуса и косинуса для каждого элемента массива x и строим два графика с использованием функции plot(). Каждый график имеет свою метку, которая будет отображаться в легенде. Наконец, мы вызываем функцию legend() без параметров, чтобы отобразить легенду на графике.

Настройка внешнего вида легенды

Вы можете настроить внешний вид легенды с помощью различных параметров функции legend(). Некоторые из наиболее полезных параметров включают:

• loc: указывает положение легенды на графике. Например, loc='upper right' поместит легенду в правый верхний угол графика.
• title: устанавливает заголовок легенды.
• fontsize: устанавливает размер шрифта легенды.
• frameon: указывает, нужно ли отображать рамку вокруг легенды.

plt.plot(x, y1, label='sin(x)')
plt.plot(x, y2, label='cos(x)')

plt.legend(loc='upper right', title='Легенда', fontsize=12, frameon=False)
plt.show()

В приведенном выше примере мы использовали параметры функции legend() для изменения внешнего вида легенды. Мы установили положение легенды в правый верхний угол графика с помощью параметра loc='upper right'. Также мы добавили заголовок легенды с помощью параметра title='Легенда', увеличили размер шрифта до 12 с помощью параметра fontsize=12 и убрали рамку вокруг легенды с помощью параметра frameon=False.

Использование собственных меток в легенде

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

plt.plot(x, y1, label='sin(x)')
plt.plot(x, y2, label='cos(x)')

plt.legend(labels=['Синус', 'Косинус'])
plt.show()

В приведенном выше примере мы установили свои метки для легенды с помощью параметра labels=['Синус', 'Косинус']. Теперь легенда будет содержать метки "Синус" и "Косинус" для соответствующих графиков.

Заключение

В этой статье мы рассмотрели, как создать легенду в matplotlib и настроить ее внешний вид. Мы узнали, как добавить метки к графикам, расположить легенду на графике, добавить заголовок и изменить размер шрифта легенды. Теперь вы можете легко добавить информативную легенду к своим графикам с использованием matplotlib.