You are reading an old version of the documentation (v1.2.1). For the latest version see

This Page

api example code: line_with_text.pyΒΆ

(Source code, png, hires.png, pdf)

Show how to override basic methods so an artist can contain another
artist.  In this case, the line contains a Text instance to label it.
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.lines as lines
import matplotlib.transforms as mtransforms
import matplotlib.text as mtext

class MyLine(lines.Line2D):

   def __init__(self, *args, **kwargs):
      # we'll update the position when the line data is set
      self.text = mtext.Text(0, 0, '')
      lines.Line2D.__init__(self, *args, **kwargs)

      # we can't access the label attr until *after* the line is
      # inited

   def set_figure(self, figure):
      lines.Line2D.set_figure(self, figure)

   def set_axes(self, axes):
      lines.Line2D.set_axes(self, axes)

   def set_transform(self, transform):
      # 2 pixel offset
      texttrans = transform + mtransforms.Affine2D().translate(2, 2)
      lines.Line2D.set_transform(self, transform)

   def set_data(self, x, y):
      if len(x):
         self.text.set_position((x[-1], y[-1]))

      lines.Line2D.set_data(self, x, y)

   def draw(self, renderer):
      # draw my label at the end of the line with 2 pixel offset
      lines.Line2D.draw(self, renderer)

fig = plt.figure()
ax = fig.add_subplot(111)
x, y = np.random.rand(2, 20)
line = MyLine(x, y, mfc='red', ms=12, label='line label')
#line.text.set_text('line label')


Keywords: python, matplotlib, pylab, example, codex (see Search examples)