Аннотировать текстовую стрелку

Введение

В этом руководстве показано, как создать нарисованную диаграмму рассеяния с текстовой стрелкой с использованием Matplotlib в Python.

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

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

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

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

Импортируем библиотеки и генерируем случайные данные

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

import matplotlib.pyplot as plt
import numpy as np

## Fixing random state for reproducibility
np.random.seed(19680801)

fig, ax = plt.subplots(figsize=(5, 5))
ax.set_aspect(1)

x1 = -1 + np.random.randn(100)
y1 = -1 + np.random.randn(100)
x2 = 1. + np.random.randn(100)
y2 = 1. + np.random.randn(100)

ax.scatter(x1, y1, color="r")
ax.scatter(x2, y2, color="g")

Добавляем текстовые аннотации на график

Далее мы добавим текстовые аннотации на график с использованием функции ax.text(). Создадим две аннотации, одну для "Sample A" и другую для "Sample B".

bbox_props = dict(boxstyle="round", fc="w", ec="0.5", alpha=0.9)
ax.text(-2, -2, "Sample A", ha="center", va="center", size=20,
        bbox=bbox_props)
ax.text(2, 2, "Sample B", ha="center", va="center", size=20,
        bbox=bbox_props)

Добавляем текстовую стрелку для указания направления

Для указания направления данных мы добавим текстовую стрелку с использованием функции ax.text() и параметра bbox с boxstyle, установленным на "rarrow".

bbox_props = dict(boxstyle="rarrow", fc=(0.8, 0.9, 0.9), ec="b", lw=2)
t = ax.text(0, 0, "Direction", ha="center", va="center", rotation=45,
            size=15,
            bbox=bbox_props)

bb = t.get_bbox_patch()
bb.set_boxstyle("rarrow", pad=0.6)

Задаем пределы графика и показываем график

Наконец, мы зададим пределы по осям x и y графика и покажем график с использованием функций ax.set_xlim(), ax.set_ylim() и plt.show().

ax.set_xlim(-4, 4)
ax.set_ylim(-4, 4)

plt.show()

Резюме

В этом уроке мы узнали, как создавать аннотированную диаграмму рассеяния с текстовой стрелкой с использованием Matplotlib в Python. Мы использовали функцию ax.text(), чтобы добавить аннотации и текстовую стрелку на график, а также функции ax.set_xlim(), ax.set_ylim() и plt.show(), чтобы задать пределы графика и показать график.