How to manually add a legend with a color box on a Matplotlib figure?

Introduction

When working with Matplotlib in Python, adding a legend with a color box can enhance the clarity of your plots, especially when dealing with multiple data series. This guide demonstrates how to manually add such a legend to your figure.

Understanding Legends in Matplotlib

A legend in Matplotlib serves as a key to your plot, explaining what each line, marker, or color represents. While Matplotlib can automatically generate legends based on plot labels, there are instances where manual control is beneficial.

Creating a Legend with a Color Box

To manually add a legend with a color box, Matplotlib provides the matplotlib.patches.Patch class. This allows you to create custom legend entries with specified colors.

Step-by-Step Example

Here's how you can implement this:

import matplotlib.pyplot as plt
import matplotlib.patches as mpatches

# Create some data
x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 2, 3, 4, 5]

# Plot the data
plt.plot(x, y1, label='Line 1')
plt.plot(x, y2, label='Line 2')

# Create color patches for the legend
patches = [mpatches.Patch(color='blue', label='Line 1'),
           mpatches.Patch(color='orange', label='Line 2')]

# Add the legend with color boxes
plt.legend(handles=patches)

# Display the plot
plt.show()

Customizing the Legend

• Position: Use the loc parameter in plt.legend() to specify the legend's location. For example, loc='upper right'.
• Title: Add a title to the legend using the title parameter.
• Frame: Control the frame around the legend with the frameon parameter. Set frameon=False to remove the frame.

Conclusion

Manually adding a legend with a color box in Matplotlib provides greater control over your plot's appearance, making it easier to convey information effectively. By using matplotlib.patches.Patch, you can customize your legends to suit your specific needs.