Matplotlib legends in subplot

To add legends in a subplot, we can take the following Steps −

• Using numpy, create points for x, y1, y2 and y3.

• Create a figure and a set of subplots, using the subplots() method, considering 3 subplots.

• Plot the curve on all the subplots(3), with different labels, colors. To place the legend for each curve or subplot adding label.

• To activate label for each curve, use the legend() method.

• To display the figure, use the show() method.

Example

import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
x = np.linspace(-2, 2, 100)
y1 = np.sin(x)
y2 = np.cos(x)
y3 = np.tan(x)
f, axes = plt.subplots(3)
axes[0].plot(x, y1, c='r', label="sine")
axes[0].legend(loc='upper left')
axes[1].plot(x, y2, c='g', label="cosine")
axes[1].legend(loc='upper left')
axes[2].plot(x, y3, c='b', label="tan")
axes[2].legend(loc='upper left')
plt.show()

Output

Rishikesh Kumar Rishi

Updated on: 26-Oct-2023

22K+ Views

Kickstart Your Career

Get certified by completing the course

Get Started