Complete Guide to Sharing a Single Colorbar for Multiple Subplots in Matplotlib

Problem Background and Challenges

In scientific computing and data visualization, it is often necessary to display multiple related data subsets in the same figure while using a unified color mapping to maintain visual consistency. Matplotlib, as one of the most popular plotting libraries in Python, provides powerful subplot and colorbar functionalities. However, users frequently encounter layout and size mismatch issues when dealing with multiple subplots sharing a single colorbar.

Core Solution

By creating an independent colorbar axis and adjusting the subplot layout, the problem of multiple subplots sharing a colorbar can be effectively resolved. Key steps include using fig.subplots_adjust() to reserve space for the colorbar, followed by creating a dedicated colorbar axis via fig.add_axes().

Detailed Implementation Steps

First, create a subplot grid and generate sample data:

import numpy as np
import matplotlib.pyplot as plt

# Create a 2x2 subplot grid
fig, axes = plt.subplots(nrows=2, ncols=2)

# Generate random data and plot heatmaps for each subplot
for ax in axes.flat:
    im = ax.imshow(np.random.random((10,10)), vmin=0, vmax=1)

Next, adjust the main plot area to reserve space for the colorbar:

# Compress the main plot area to the right to make space for the colorbar
fig.subplots_adjust(right=0.8)

Create an independent colorbar axis and add the colorbar:

# Create colorbar axis on the right side, parameters are [left, bottom, width, height]
cbar_ax = fig.add_axes([0.85, 0.15, 0.05, 0.7])

# Associate colorbar with the last image object
fig.colorbar(im, cax=cbar_ax)

plt.show()

Importance of Data Normalization

To ensure all subplots use the same color mapping range, data normalization must be unified. When different subplots have significantly varying data ranges, use identical vmin and vmax parameters, or create a shared Normalize object:

from matplotlib import colors

# Create a shared normalization object
norm = colors.Normalize(vmin=0, vmax=1)

for ax in axes.flat:
    im = ax.imshow(np.random.random((10,10)), norm=norm)

Layout Adjustment Techniques

By finely tuning the parameters of subplots_adjust and the position of the colorbar axis, optimal visual effects can be achieved. Key parameters include:

• left, right, bottom, top: Control the boundaries of the main plot area
• wspace, hspace: Control horizontal and vertical spacing between subplots

Practical Application Example

Below is a complete application example demonstrating how to create a shared colorbar for physical simulation data:

import numpy as np
import matplotlib.pyplot as plt

# Define physical functions
def gravitational_potential(x, y, mass=1.0):
    r = np.sqrt(x**2 + y**2)
    return -mass / np.where(r == 0, 1e-10, r)

# Generate coordinate grid
coords = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(coords, coords)

# Calculate gravitational potential for different masses
potential1 = gravitational_potential(X, Y, mass=1.0)
potential2 = gravitational_potential(X, Y, mass=2.0)

# Create subplots
fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(12, 5))

# Determine unified data range
vmin = min(np.min(potential1), np.min(potential2))
vmax = max(np.max(potential1), np.max(potential2))

# Plot subplots
im1 = ax1.imshow(potential1, extent=(-5, 5, -5, 5), vmin=vmin, vmax=vmax)
ax1.set_title("Mass = 1.0")
ax1.set_xlabel("x coordinate")
ax1.set_ylabel("y coordinate")

im2 = ax2.imshow(potential2, extent=(-5, 5, -5, 5), vmin=vmin, vmax=vmax)
ax2.set_title("Mass = 2.0")
ax2.set_xlabel("x coordinate")
ax2.set_ylabel("y coordinate")

# Adjust layout and add colorbar
fig.subplots_adjust(right=0.85)
cbar_ax = fig.add_axes([0.88, 0.15, 0.02, 0.7])
fig.colorbar(im1, cax=cbar_ax, label="Gravitational Potential")

plt.tight_layout()
plt.show()

Common Issues and Solutions

In practical applications, the following common issues may arise:

Issue 1: Colorbar and subplot size mismatch

Solution: Precisely control the layout by adjusting the position parameters of add_axes and the boundary parameters of subplots_adjust.

Issue 2: Large data range differences between subplots

Solution: Ensure color mapping consistency using shared Normalize objects or unified vmin/vmax parameters.

Issue 3: Colorbar label overlap

Solution: Adjust the size of the colorbar axis or use the labelpad parameter to increase label spacing.

Advanced Techniques

For more complex layout requirements, consider using Matplotlib's GridSpec to create more flexible subplot arrangements:

import matplotlib.gridspec as gridspec

fig = plt.figure(figsize=(10, 8))
gs = gridspec.GridSpec(2, 2, width_ratios=[4, 1], height_ratios=[1, 1])

# Create main plot areas
ax1 = fig.add_subplot(gs[0, 0])
ax2 = fig.add_subplot(gs[1, 0])

# Create colorbar area
cax = fig.add_subplot(gs[:, 1])

# Plot images and colorbar
im1 = ax1.imshow(np.random.random((10, 10)))
im2 = ax2.imshow(np.random.random((10, 10)))
fig.colorbar(im1, cax=cax)

plt.show()

Conclusion

Through the methods introduced in this article, users can effectively create shared colorbars for multiple subplots in Matplotlib. The key lies in understanding Matplotlib's axis system and layout mechanisms, as well as the importance of data normalization. These techniques are applicable not only to simple subplot layouts but can also be extended to more complex visualization scenarios.