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

Введение

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

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

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

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

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

Импорт Matplotlib

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

import matplotlib.pyplot as plt

## Create a simple plot
fig, ax = plt.subplots()
ax.plot([0, 1, 2, 3, 4], [0, 1, 4, 9, 16])
plt.show()

Добавление текстовой аннотации

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

## Add text annotation
ax.text(2, 10, "Important Point", fontsize=12, color='red')
plt.show()

Добавление стрелковой аннотации

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

## Find the maximum value
y = [0, 1, 4, 9, 16]
max_index = y.index(max(y))
xmax = max_index
ymax = y[max_index]

## Add arrow annotation
ax.annotate('Maximum Value', xy=(xmax, ymax), xytext=(xmax, ymax + 5),
            arrowprops=dict(facecolor='black', shrink=0.05))
plt.show()

Добавление аннотации в виде фигуры

Фигуры можно использовать, чтобы привлечь внимание к определенным областям графика. В этом шаге мы добавим прямоугольник, чтобы выделить область между x = 1 и x = 3.

## Add shape annotation
ax.axvspan(1, 3, facecolor='gray', alpha=0.2)
plt.show()

Резюме

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