Label a whole numpy array with one label on matplotlib

Question

I would like to label a whole numpy array with only one label. The following code for example creates 6 (=2+4) labels instead of only 2 labels:

import numpy as np
import matplotlib.pyplot as plt

a = np.random.rand(10,2)
b = np.random.rand(10,4)


plt.figure()
plt.plot(a, 'blue', label = 'a')
plt.plot(b, 'red', label = 'b')
plt.legend()

How should the code above be modified to create only 2 legend labels, 'a' and 'b'?

Could you clarify how many lines you expect to be drawn from the array a or b? — medium-dimensional
– medium-dimensional, Commented Nov 21, 2022 at 6:42
@medium-dimensional I would like to get one label per numpy.array, or one label per color, while plotting 2 blue lines (labeled a ) and 4 red lines (labeled b), i.e. the graph should stay the same, the legend should be only one legend blue line - a, one legend red line - b — Nir
– Nir, Commented Nov 21, 2022 at 11:13

Joao_PS · Accepted Answer · 2022-11-21 08:55:19Z

2

a_lines = plt.plot(a, c='blue')
b_lines = plt.plot(b, c='red')
plt.legend(handles=[a_lines[0], b_lines[0]], labels=['a', 'b'])

answered Nov 21, 2022 at 8:55

Joao_PS

6705 silver badges10 bronze badges

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

Collectives™ on Stack Overflow

Label a whole numpy array with one label on matplotlib

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