Я работаю над реализацией класса для встроенной маркировки объектов линии. Для этого я создал подкласс класса Text, который как объект Line2D как атрибут. Код в предыдущем сообщении был, возможно, немного длинным, поэтому я выделил проблему здесь:
from matplotlib.text import Text
from matplotlib import pyplot as plt
import numpy as np
class LineText(Text):
def __init__(self,line,*args,**kwargs):
x_pos = line.get_xdata().mean()
y_pos = line.get_ydata().mean()
Text.__init__(self,x=x_pos,y=y_pos,*args,**kwargs)
self.line = line
def draw(self,renderer):
self.line.set_color(self.get_color())
self.line.draw(renderer = renderer)
Text.draw(self,renderer)
if __name__ == '__main__':
x = np.linspace(0,1,20)
y = np.linspace(0,1,20)
ax = plt.subplot(1,1,1)
line = plt.plot(x,y,color = 'r')[0]
linetext = LineText(line,text = 'abc')
ax.add_artist(linetext)
plt.show()
Класс берет дескриптор Line2D как возвращаемый функцией plot и в методе .draw, он вносит некоторые изменения в строку. В целях иллюстрации я здесь просто попытался изменить свой цвет.
После изменения цвета строки я вызываю строки draw. Однако это не имеет ожидаемого эффекта. Когда рисунок сначала рисуется, кажется, что наложение красной и черной линий. Как только размер будет изменен или будет принудительно перерисован, линия изменит свой цвет, как ожидалось. Единственный способ, который я нашел до сих пор, чтобы заставить рисунок правильно рисовать при открытии, заключался в том, чтобы добавить plt.draw() до show(). Это, однако, кажется неуклюжим.
Могу ли я каким-то образом изменить только объект линии, который нужно перерисовать? Или я делаю это совершенно неправильно?
Спасибо заранее.
