Аннотирование графика с использованием Matplotlib

Введение

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

Советы по работе с ВМ

После запуска ВМ нажмите в левом верхнем углу, чтобы переключиться на вкладку Notebook и получить доступ к Jupyter Notebook для практики.

Иногда вам может потребоваться подождать несколько секунд, пока Jupyter Notebook не загрузится полностью. Валидация операций не может быть автоматизирована из-за ограничений Jupyter Notebook.

Если вы сталкиваетесь с проблемами во время обучения, не стесняйтесь обращаться к Labby. Оставьте отзыв после занятия, и мы оперативно решим проблему для вас.

Импортировать библиотеки

Прежде чем мы начнем, нам нужно импортировать необходимые библиотеки. В этом руководстве мы будем использовать Matplotlib и Numpy.

import matplotlib.pyplot as plt
import numpy as np

Создать график

Далее мы создадим график с использованием Matplotlib. В этом примере мы построим функцию косинуса на заданном диапазоне значений.

fig, ax = plt.subplots()

t = np.arange(0.0, 5.0, 0.01)
s = np.cos(2*np.pi*t)
line, = ax.plot(t, s, lw=2)

Аннотировать график

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

ax.annotate('local max', xy=(2, 1), xytext=(3, 1.5),
            arrowprops=dict(facecolor='black', shrink=0.05),
            )

Функция ax.annotate() принимает несколько аргументов. Первый аргумент - это текст, который будет отображаться на графике. Аргумент xy задает координаты точки, которую мы хотим аннотировать. Аргумент xytext задает координаты текста. Аргумент arrowprops - это словарь, который задает свойства стрелки.

Установить пределы графика

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

ax.set_ylim(-2, 2)
plt.show()

Резюме

В этом руководстве мы научились аннотировать график с использованием Matplotlib. Мы начали с импорта необходимых библиотек и создания графика. Затем мы аннотировали график, добавив стрелку, указывающую на конкретные координаты. Наконец, мы установили пределы графика, чтобы убедиться, что отмеченная точка видна.