Как я могу сделать метки xtick на графике простыми чертежами с использованием matplotlib? - PullRequest
Новая
       42

Как я могу сделать метки xtick на графике простыми чертежами с использованием matplotlib?

13 голосов
/ 05 января 2012

Вместо слов или цифр, являющихся метками тиков оси x, я хочу нарисовать простой рисунок (сделанный из линий и окружностей) в качестве метки для каждого тика x. Это возможно? Если да, то как лучше всего это сделать в matplotlib?

Ответы [ 2 ]

25 голосов
/ 05 января 2012

Я бы убрал галочки и заменил текст на патчи . Вот краткий пример выполнения этой задачи: 

import matplotlib.pyplot as plt
import matplotlib.patches as patches


# define where to put symbols vertically
TICKYPOS = -.6

fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(range(10))

# set ticks where your images will be
ax.get_xaxis().set_ticks([2,4,6,8])
# remove tick labels
ax.get_xaxis().set_ticklabels([])


# add a series of patches to serve as tick labels
ax.add_patch(patches.Circle((2,TICKYPOS),radius=.2,
                            fill=True,clip_on=False))
ax.add_patch(patches.Circle((4,TICKYPOS),radius=.2,
                            fill=False,clip_on=False))
ax.add_patch(patches.Rectangle((6-.1,TICKYPOS-.05),.2,.2,
                               fill=True,clip_on=False))
ax.add_patch(patches.Rectangle((8-.1,TICKYPOS-.05),.2,.2,
                               fill=False,clip_on=False))

Это приводит к следующему рисунку:

enter image description here

Ключом для установки clip_on является False, иначе patches вне осей не будет отображаться. Координаты и размеры (радиус, ширина, высота и т. Д.) Участков будут зависеть от того, где находятся ваши оси на рисунке. Например, если вы планируете делать это с помощью вспомогательных участков, вам нужно будет учитывать размещение исправлений, чтобы не перекрывать другие оси. Возможно, стоит потратить время на изучение Преобразований и определение позиций и размеров в другой единице (Оси, Рисунок или дисплей).

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

import matplotlib.pyplot as plt

fig = plt.figure(figsize=(1,1),dpi=400)
ax = fig.add_axes([0,0,1,1],frameon=False)
ax.set_axis_off()

ax.plot(range(10),linewidth=32)
ax.plot(range(9,-1,-1),linewidth=32)

fig.savefig('thumb.png')

производит это изображение:

enter image description here

Затем я создал BboxImage в том месте, где хочу метку галочки, и нужного размера: 

lowerCorner = ax.transData.transform((.8,TICKYPOS-.2))
upperCorner = ax.transData.transform((1.2,TICKYPOS+.2))

bbox_image = BboxImage(Bbox([lowerCorner[0],
                             lowerCorner[1],
                             upperCorner[0],
                             upperCorner[1],
                             ]),
                       norm = None,
                       origin=None,
                       clip_on=False,
                       )

Заметил, как я использовал преобразование transData для преобразования единиц измерения в единицы измерения, которые требуются в определении Bbox.

Теперь я читаю на изображении, используя подпрограмму imread, и устанавливаю его результаты (массив NumPy) для данных bbox_image и добавляю художника к осям: 

bbox_image.set_data(imread('thumb.png'))
ax.add_artist(bbox_image)

Это приводит к обновленной фигуре: enter image description here

Если вы используете изображения напрямую, обязательно импортируйте необходимые классы и методы: 

from matplotlib.image import BboxImage,imread
from matplotlib.transforms import Bbox
4 голосов
/ 20 сентября 2018

Другой ответ имеет некоторые недостатки, поскольку использует статические координаты. Следовательно, он не будет работать при изменении размера фигуры или масштабировании и панорамировании графика.

Лучшим вариантом является непосредственное определение позиций в выбранных системах координат. Для оси X имеет смысл использовать координаты данных для позиции x и координаты осей для позиции y.

Использование matplotlib.offsetbox es делает это довольно простым. Следующее будет располагать прямоугольник с кружком и прямоугольник с изображением в координатах (-5,0) и (5,0) соответственно и смещать их немного в нижнее положение, чтобы они выглядели так, как будто они были метками. 

import matplotlib.pyplot as plt
import matplotlib.patches as mpatches
from matplotlib.offsetbox import (DrawingArea, OffsetImage,AnnotationBbox)

fig, ax = plt.subplots()
ax.plot([-10,10], [1,3])

# Annotate the 1st position with a circle patch
da = DrawingArea(20, 20, 10, 10)
p = mpatches.Circle((0, 0), 10)
da.add_artist(p)

ab = AnnotationBbox(da, (-5,0),
                    xybox=(0, -7),
                    xycoords=("data", "axes fraction"),
                    box_alignment=(.5, 1),
                    boxcoords="offset points",
                    bboxprops={"edgecolor" : "none"})

ax.add_artist(ab)

# Annotate the 2nd position with an image
arr_img = plt.imread("https://i.stack.imgur.com/FmX9n.png", format='png')

imagebox = OffsetImage(arr_img, zoom=0.2)
imagebox.image.axes = ax

ab = AnnotationBbox(imagebox, (5,0),
                    xybox=(0, -7),
                    xycoords=("data", "axes fraction"),
                    boxcoords="offset points",
                    box_alignment=(.5, 1),
                    bboxprops={"edgecolor" : "none"})

ax.add_artist(ab)

plt.show()

enter image description here

Обратите внимание, что многие формы существуют как символы Юникода, так что можно просто установить метки с этими символами. О таком решении см. Как использовать цветную фигуру в качестве утиков в matplotlib или seaborn?

...