How to add titles to the legend rows in Matplotlib?

To add titles to the legend rows in Matplotlib, we can take the following steps −

Set the figure size and adjust the padding between and around the subplots.
Create y data points using numpy.
Make lists of markers and labels.
Create a figure and a set of subplots.
Plot the lines using plot() method, with different labels and markers.
Get the plot handlers for half of the plot.
Get the labels for the legends.
Place the legends on the plot.
To display the figure, use show() method.

Example

import matplotlib.pyplot as plt
import numpy as np

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

y = np.exp(-np.arange(5))

markers = ["s", "o", "*"]
labels = ["one", "two"]

fig, ax = plt.subplots()

for i in range(6):
   ax.plot(y * i + i / 2., marker=markers[i // 2], label=labels[i % 2])
h, l = ax.get_legend_handles_labels()
ph = [plt.plot([], marker="", ls="")[0]] * 2
handles = ph + h

labels = ["Title 1:", "Title 2:"] + l
plt.legend(handles, labels, ncol=4)

plt.show()

Output

Rishikesh Kumar Rishi

Updated on: 2021-08-10T06:52:57+05:30

2K+ Views

Kickstart Your Career

Get certified by completing the course

Get Started