python - Show the final y-axis value of each line with matplotlib

Question

I'm drawing a graph with some lines using matplotlib and I want to display the final y value next to where each line ends on the right hand side like this:

Any solutions or pointers to the relevant parts of the API? I'm quite stumped.

I'm using matplotlib 1.0.0 and the pyplot interface, e.g. pyplot.plot(xs, ys, f, xs_, ys_, f_).

Answer 1

While there's nothing wrong with Ofri's answer, annotate is intended especially for this purpose:

import matplotlib.pyplot as plt
import numpy as np

x = np.arange(61).astype(np.float)
y1 = np.exp(0.1 * x)
y2 = np.exp(0.09 * x)

plt.plot(x, y1)
plt.plot(x, y2)

for var in (y1, y2):
    plt.annotate('%0.2f' % var.max(), xy=(1, var.max()), xytext=(8, 0), 
                 xycoords=('axes fraction', 'data'), textcoords='offset points')

plt.show()

This places the text 8 points to the right of the right side of the axis, at the maximum y-value for each plot. You can also add in arrows, etc. See http://matplotlib.sourceforge.net/users/annotations_guide.html (You can also change the vertical alignment, if you want the text vertically centered on the given y-value. Just specify va='center'.)

Also, this doesn't rely on tick locations, so it will work perfectly for log plots, etc. Giving the location of the text in terms of the positions of the axis boundaries and its offset in points has a lot of advantages if you start rescaling the plot, etc.