Adding a legend to a Matplotlib boxplot with multiple plots on the same axis

MatplotlibPythonData Visualization

To add a legend to a matplotlib boxplot with multiple plots on the same axis, we can take the following steps −

  • Set the figure size and adjust the padding between and around the subplots.

  • Create random data, a and b, using numpy.

  • Create a new figure or activate an existing figure using figure() method.

  • Add an axes to the current figure as a subplot arrangement.

  • Make a box and whisker plot using boxplot() method with different facecolors.

  • To place the legend, use legend() method with two boxplots, bp1 and bp2, and ordered label for legend elements.

  • 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

a = np.random.rand(100, 2)
b = np.random.rand(100, 2)

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

bp1 = ax.boxplot(a, positions=[1, 3], notch=True, widths=0.35, patch_artist=True, boxprops=dict(facecolor="C0"))
bp2 = ax.boxplot(a, positions=[0, 2], notch=True, widths=0.35, patch_artist=True, boxprops=dict(facecolor="C2"))
ax.legend([bp1["boxes"][0], bp2["boxes"][0]], ["Box Plot 1", "Box Plot 2"], loc='upper right')

plt.show()

Output

raja
Published on 03-Jun-2021 13:12:09
Advertisements