Adding legends to plots when not using the pyplot interface

Question

Say I have the following:

fig  = figure()
ax   = f.add_subplot(111)

# my_values holds a 2D numpy array
lines = ax.plot(my_values)

Say that for each column of my_values I have a string holding the legend that I want associated with the corresponding line.

e.g.

my_legends = ['foo ' + str(x) for x in xrange(my_values.shape[1])]

I have the handles for the figure (fig), axes (ax) and the lines (lines), how can I add these legends to the plot?

unutbu · Accepted Answer · 2014-03-26 20:54:45Z

2

You could call ax.legend:

ax.legend(lines, my_legends)

import matplotlib.pyplot as plt
import numpy as np

fig  = plt.figure()
ax   = fig.add_subplot(111)
my_values = np.cumsum(np.random.random(100)-0.5).reshape(-1, 2)
lines = ax.plot(my_values)
ax.legend(lines, ['eenie', 'meenie'], loc='best')
plt.show()

enter image description here

edited Mar 26, 2014 at 20:54

answered Mar 26, 2014 at 20:49

unutbu

886k197 gold badges1.9k silver badges1.7k bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Adding legends to plots when not using the pyplot interface

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related