add legend to numpy array in matplot lib

Question

I am plotting 2D numpy arrays using

import numpy as np
import matplotlib.pyplot as plt

x = np.array([1,2,3])
y = np.array([[2,2.2,3],[1,5,1]])
plt.plot(x,y.T[:,:])
plt.legend()
plt.show()

I want a legend that tells which line belongs to which row. Of course, I realize I can't give it meaningful names, but I need some sort of unique label for the line without running through loop.

tacaswell · Accepted Answer · 2016-04-30 19:16:46Z

2

import numpy as np
import matplotlib.pyplot as plt
import uuid

x = np.array([1,2,3])
y = np.array([[2,2.2,3],[1,5,1]])

fig, ax = plt.subplots()
lines = ax.plot(x,y.T[:,:])
ax.legend(lines, [str(uuid.uuid4())[:6] for j in range(len(lines))])


plt.show()

(This is off of the current mpl master branch with a preview of the 2.0 default styles)

answered Apr 30, 2016 at 19:16

tacaswell

88.2k23 gold badges222 silver badges203 bronze badges

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

2 Comments

tacaswell Over a year ago

The OP asked for unique labels, so those labels are guaranteed to be unique ;)

DeanLa Over a year ago

ued [ j+1 for j in range(len(lines))] which is unique enough. Thank you

Collectives™ on Stack Overflow

add legend to numpy array in matplot lib

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related