Использование библиотеки Matplotlib. Как рисовать стрелки на графиках и добавлять аннотации

Иногда на графиках нужно нарисовать указатель на какую-то особую точку, на которую стоит обратить внимание, или указательные линии с номерами кривых могут служить заменой легенды. Для того, чтобы рисовать такие стрелки и линии, предназначена функция annotate. Кроме того, эту функцию можно использовать для отображения векторов на плоскости.

Функция annotate может принимать большое количество параметров, отвечающих за внешний вид стрелок и отображаемого текста. Строго говоря, обязательными являются только два параметра:

s - строка, содержащая выводимый текст;
xy - кортеж из двух элементов, задающие координаты точки, на которую нужно указать. По умолчанию координаты точек задаются в системе координат графика, но это поведение можно менять с помощью параметра xycoords, который в данной статье не рассматривается.

Однако, если мы ограничимся только этими параметрами, то функция annotate выведет только текст около указанной точки:

# -*- coding: UTF-8 -*-

import pylab
import numpy

if __name__ == '__main__':
def sinc (x):
return numpy.sin (x) / x

# Интервал изменения и шаг переменной по оси X
xmin = -30.0
xmax = 30.0
dx = 0.01

# Создадим список координат по осям X и Y на отрезке [-xmin; xmax], включая концы
x = numpy.arange (xmin, xmax, dx)
y = sinc(x)

# Нарисуем график
pylab.plot (x, y)

# Координаты точки, куда будет указывать стрелка
selected_x = 7.7
selected_y = sinc (selected_x)

# Установим шрифт, который может отображать русские буквы
pylab.rc('font', family = 'verdana')

# !!! Добавление аннотации
pylab.annotate (u'Очень важная точка',
xy=(selected_x, selected_y))

# Покажем окно с нарисованным графиком
pylab.show()

Исходник

Результат будет напоминать работу функции text:

Чтобы аннотация выглядела более интересно, добавим два именованных параметра:

xytext - кортеж из двух элементов, который задает координаты текста; из этой же точки будет начинаться стрелка указателя;
arrowprops - словарь, который содержит элементы, задающие внешний вид стрелки.

Чуть позже мы подробнее рассмотрим возможные значения в словаре arrowprops, а пока сделаем так, чтобы у нас отображалась обычная стрелка:

# -*- coding: UTF-8 -*-

import pylab
import numpy

if __name__ == '__main__':
def sinc (x):
return numpy.sin (x) / x

# Интервал изменения и шаг переменной по оси X
xmin = -30.0
xmax = 30.0
dx = 0.01

# Создадим список координат по осям X и Y на отрезке [-xmin; xmax], включая концы
x = numpy.arange (xmin, xmax, dx)
y = sinc(x)

# Нарисуем график
pylab.plot (x, y)

# Координаты точки, куда будет указывать стрелка
selected_x = 7.7
selected_y = sinc (selected_x)

pylab.rc('font', family = 'verdana')

# !!! Словарь задает внешний вид стрелки
arrowprops = {
'arrowstyle': '->',
}

# !!! Добавление аннотации
pylab.annotate (u'Очень важная точка',
xy=(selected_x, selected_y),
xytext = (selected_x + 2, selected_y + 0.1),
arrowprops = arrowprops)

# Покажем окно с нарисованным графиком
pylab.show()

Исходник

Теперь окно с графиком будет выглядеть следующим образом:

Если говорить о внутреннем устройстве Matplotlib, то функция annotate для отображения стрелок использует один из двух классов: FancyArrowPatch, если словарь arrowprops содержит ключ "arrowstyle" (как в предыдущем примере), или класс YAArrow, если такого ключа в словаре нет. На мой взгляд стрелки, которые рисуются с помощью FancyArrowPatch, по умолчанию выглядят более привлекательно.

С помощью элементов словаря arrowprops мы можем передавать множество настроек в класс FancyArrowPatch. Коротко рассмотрим только некоторые из них:

arrowstyle - задает стиль стрелок (список возможных значений будет приведен ниже).
connectionstyle - задает способ проведения линии указателя, будет она соединять две точки по прямой, по дуге или нужно, чтобы стрелки состояли из ломанных отрезков, расположенных параллельно координатным осям. С помощью этого параметра можно задавать и другие настройки линии.
linewidth - задает толщину линии.
linestyle - задает стиль линии. Возможные значения - 'solid', 'dashed', 'dashdot', '‘dotted'.
mutation_scale - задает размер "головы" стрелки на конце.

Если словарь arrowprops не содержит ключ "arrowstyle", то для отображения стрелки будет использоваться класс YAArrow. Этот класс будет использовать следующие ключи в словаре arrowprops:

width - задает ширину линии указателя.
frac - коэффициент, показывающий, какую часть длины указателя будет занимать "голова" стрелки. Например, если frac = 0.5, то стрелка будет занимать половину длины указателя.
headwidth - задает ширину "головы" стрелки.

Кроме того, независимо от используемого класса FancyArrowPatch / YAArrow, можно задавать следующие параметры:

color - задает цвет указателя.
alpha - задает степень прозрачности. 0 - полностью прозрачный, 1 - полностью непрозрачный.

Если используется параметр arrowstyle, то он может быть экземпляром класса ArrowStyle, который позволяет очень гибко настраивать внешний вид стрелок, а может быть строкой, описывающей уже предустановленные стили стрелок. В качестве строки arrowstyle может принимать следующие значения:

'-'

'->'

'-['

'-|>'

'<-'

'<->'

'<|-'

'-'

']-'

']-['

'fancy'

'simple'

'wedge'

'|-|'

Чтобы продемонстрировать, как выглядят эти стили, был написан следующий скрипт:

# -*- coding: UTF-8 -*-

import pylab

if __name__ == '__main__':
pylab.rc('font', family = 'verdana')
pylab.xlim ([-10, 50])
pylab.ylim ([-180, 20])

arrowstyles = ['-', '->', '-[', '-|>',
'<-', '<->', '<|-', '<|-|>',
']-', ']-[', 'fancy', 'simple',
'wedge', '|-|']

shift_x = 2
shift_y = -12

dx = 10
dy = 10

for n, arrowstyle in enumerate (arrowstyles):
# !!! Словарь задает внешний вид стрелки
arrowprops = {
'arrowstyle': arrowstyle,
}

x = n * shift_x
y = n * shift_y

# !!! Добавление аннотации
pylab.annotate (u'arrowstyle: ' + arrowstyle,
xy=(x, y),
xytext = (x + dx, y + dy),
arrowprops = arrowprops)

# Покажем окно с нарисованным графиком
pylab.show()

Исходник

С помощью него была получена следующая картинка:

Кроме того, мы можем влиять на то, каким образом будет проводиться линия от начальной точки до конечной: будет это прямая линия, ломаная или закругленная. На это влияет параметр connectionstyle, который может быть экземпляром класса ConnectionStyle для очень точной настройки внешнего вида, а может быть одной из строк, с помощью которых можно задать один из предустановленных внешних видов указателей. Вот эти возможные строки:

'angle'

'angle3'

'arc'

'arc3'

'bar'

Цифра "3" в конце некоторых стилей обозначают, что для проведения линии используется квадратичный сплайн, то есть линия проводится через три точки. Чтобы продемонстрировать эти стили, был написан следующий скрипт:

# -*- coding: UTF-8 -*-

import pylab

if __name__ == '__main__':
pylab.rc('font', family = 'verdana')
pylab.xlim ([-10, 50])
pylab.ylim ([-190, 40])

connectionstyle = ['angle', 'angle3', 'arc', 'arc3', 'bar']

shift_x = 2
shift_y = -45

dx = 10
dy = 20

for n, style in enumerate (connectionstyle):
# !!! Словарь задает внешний вид стрелки
arrowprops = {
'arrowstyle': '->',
'connectionstyle': style,
}

x = n * shift_x
y = n * shift_y

# !!! Добавление аннотации
pylab.annotate (u'connectionstyle: ' + style,
xy=(x, y),
xytext = (x + dx, y + dy),
arrowprops = arrowprops)

# Покажем окно с нарисованным графиком
pylab.show()

Исходник

Результат его работы выглядит следующим образом:

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

'angle'	angleA=90,angleB=0,rad=0.0
'angle3'	angleA=90,angleB=0
'arc'	angleA=0,angleB=0,armA=None,armB=None,rad=0.0
'arc3'	rad=0.0
'bar'	armA=0.0,armB=0.0,fraction=0.3,angle=None

Для демонстрации того, как задаются параметры стилей, изменим предыдущий пример:

# -*- coding: UTF-8 -*-

import pylab

if __name__ == '__main__':
pylab.rc('font', family = 'verdana')
pylab.xlim ([-10, 50])
pylab.ylim ([-190, 40])

connectionstyle = ['angle,angleA=0,angleB=45',
'angle3,angleA=45,angleB=0',
'arc,angleA=-90,armA=30,armB=80',
'arc3,rad=0.5',
'bar,armA=15,fraction=0.1']

shift_x = 2
shift_y = -45

dx = 10
dy = 20

for n, style in enumerate (connectionstyle):
# !!! Словарь задает внешний вид стрелки
arrowprops = {
'arrowstyle': '->',
'connectionstyle': style,
}

x = n * shift_x
y = n * shift_y

# !!! Добавление аннотации
pylab.annotate (u'connectionstyle: ' + style,
xy=(x, y),
xytext = (x + dx, y + dy),
arrowprops = arrowprops)

# Покажем окно с нарисованным графиком
pylab.show()

Исходник

Не будем приводить здесь описание каждого параметра, потому что проще всего их "пощупать" самим, чтобы понять, что они делают. Результат данного скрипта выглядит следующим образом:

Функцию annotate можно использовать для отображения векторов на плоскости, для этого достаточно в качестве первого параметра передать пустую строку.

# -*- coding: UTF-8 -*-

import pylab
import numpy

if __name__ == '__main__':
pylab.rc('font', family = 'verdana')
pylab.xlim ([-1, 10])
pylab.ylim ([-1, 10])

a = numpy.array ([1, 3])
b = numpy.array ([5, 2])

c = a + b

arrowprops = {
'arrowstyle': '<|-',
}

# !!! Добавление аннотации
pylab.annotate (u'',
xy=(0, 0),
xytext = a,
arrowprops = arrowprops)

pylab.annotate (u'',
xy=(0, 0),
xytext = b,
arrowprops = arrowprops)

pylab.annotate (u'',
xy=(0, 0),
xytext = c,
arrowprops = arrowprops)

# Покажем окно с нарисованным графиком
pylab.grid()
pylab.show()

Исходник

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

# -*- coding: UTF-8 -*-

import pylab
import numpy

if __name__ == '__main__':
pylab.rc('font', family = 'verdana')
pylab.xlim ([-1, 10])
pylab.ylim ([-1, 10])

a = numpy.array ([1, 3])
b = numpy.array ([5, 2])

c = a + b

arrowprops = {
'arrowstyle': '<|-',
}

# !!! Добавление аннотации
pylab.annotate (u'$\\vec a$',
xy=(0, 0),
xytext = a,
arrowprops = arrowprops)

pylab.annotate (u'$\\vec b$',
xy=(0, 0),
xytext = b,
arrowprops = arrowprops)

pylab.annotate (u'$\\vec c = \\vec a + \\vec b$',
xy=(0, 0),
xytext = c,
arrowprops = arrowprops)

# Покажем окно с нарисованным графиком
pylab.grid()
pylab.show()

Исходник

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

Другие статьи про Matplotlib

Вы можете подписаться на новости сайта через RSS, Группу Вконтакте или Канал в Telegram.
4.5 stars

Рейтинг 4.4/5. Всего 5 голос(а, ов)

	Подписаться на комментарии
Автор:
Тема:
	Ваш комментарий


Введите код 798

jenyay.net

Софт, исходники и фото

Использование библиотеки Matplotlib. Как рисовать стрелки на графиках и добавлять аннотации