Complete Guide to Matplotlib Scatter Plot Legends: From 2D to 3D Visualization

Problem Background and Core Challenges

Creating scatter plots with clear legends is a common requirement in data visualization projects. Users often encounter legend display anomalies when using Matplotlib, particularly when handling multidimensional data. The original code uses Line2D objects to create legends, which correctly display symbols and colors but add unwanted lines, affecting visual quality.

Correct Implementation for 2D Scatter Plots

For standard 2D scatter plots, using the plt.scatter method is the most straightforward approach. The key lies in proper legend parameter configuration:

import matplotlib.pyplot as plt
from numpy.random import random

colors = ['b', 'c', 'y', 'm', 'r']

# Create multiple scatter plot series
lo = plt.scatter(random(10), random(10), marker='x', color=colors[0])
ll = plt.scatter(random(10), random(10), marker='o', color=colors[0])
l  = plt.scatter(random(10), random(10), marker='o', color=colors[1])
a  = plt.scatter(random(10), random(10), marker='o', color=colors[2])
h  = plt.scatter(random(10), random(10), marker='o', color=colors[3])
hh = plt.scatter(random(10), random(10), marker='o', color=colors[4])
ho = plt.scatter(random(10), random(10), marker='x', color=colors[4])

# Key configuration: use scatterpoints instead of numpoints
plt.legend((lo, ll, l, a, h, hh, ho),
           ('Low Outlier', 'LoLo', 'Lo', 'Average', 'Hi', 'HiHi', 'High Outlier'),
           scatterpoints=1,
           loc='lower left',
           ncol=3,
           fontsize=8)

plt.show()

This approach correctly displays legends with only one marker per legend entry, avoiding additional lines.

Special Handling for 3D Scatter Plots

When extending to 3D visualization, the situation becomes more complex. Matplotlib's 3D scatter plots return Patch3DCollection objects, which are not supported by the legend system:

import matplotlib.pyplot as plt
from numpy.random import random
from mpl_toolkits.mplot3d import Axes3D

colors=['b', 'c', 'y', 'm', 'r']

ax = plt.subplot(111, projection='3d')

# Use plot method instead of scatter method
ax.plot(random(10), random(10), random(10), 'x', color=colors[0], label='Low Outlier')
ax.plot(random(10), random(10), random(10), 'o', color=colors[0], label='LoLo')
ax.plot(random(10), random(10), random(10), 'o', color=colors[1], label='Lo')
ax.plot(random(10), random(10), random(10), 'o', color=colors[2], label='Average')
ax.plot(random(10), random(10), random(10), 'o', color=colors[3], label='Hi')
ax.plot(random(10), random(10), random(10), 'o', color=colors[4], label='HiHi')
ax.plot(random(10), random(10), random(10), 'x', color=colors[4], label='High Outlier')

plt.legend(loc='upper left', numpoints=1, ncol=3, fontsize=8, bbox_to_anchor=(0, 0))
plt.show()

This method creates 3D scatter effects through the plot method while supporting normal legend display.

Automated Legends in Modern Matplotlib

In newer Matplotlib versions (3.1.1+), more intelligent legend creation methods are available:

import matplotlib.pyplot as plt
import numpy as np

# Sample data
N = 45
x, y = np.random.rand(2, N)
c = np.random.randint(1, 5, size=N)

fig, ax = plt.subplots()
scatter = ax.scatter(x, y, c=c)

# Automatically create color-based legend
legend = ax.legend(*scatter.legend_elements(), 
                   loc="lower left", 
                   title="Classes")
plt.show()

This approach automatically identifies categories in the data and generates corresponding legends, significantly reducing code complexity.

Technical Key Points Summary

Understanding the core mechanisms of Matplotlib's legend system is crucial:

• 2D Scatter Plots: Use plt.scatter with scatterpoints=1 parameter
• 3D Scatter Plots: Must use ax.plot method because objects returned by Axes3D.scatter are not legend-supported
• Modern Methods: Utilize legend_elements() for automated legend creation
• Parameter Selection: Correct usage of scatterpoints vs numpoints depends on the plotting method used

Best Practice Recommendations

In practical projects, the following strategies are recommended:

1. Clarify data dimensions and select corresponding plotting methods
2. For complex legend requirements, consider using loop structures to simplify code
3. Leverage Matplotlib's automation features to reduce manual configuration
4. Test compatibility across different versions, especially when using new features

By mastering these technical points, developers can create both aesthetically pleasing and functionally complete scatter plot visualizations.