Remove colorbar from plot_topomap()

đź’¬ Support & Discussions
1

Hi,
I am not able to remove the colorbar from the “epochs.compute_psd().plot_topomap()” function even if i set colorbar=False. I want to plot two topo maps in one figure with one common colorbar, but can’t remove the colorbar for each frequency band topo map.

This is my code:

channels = ["Fp1", "AF7", "AF8", "FC4", "FC5", "FC6", "T7", "Cz", "PO7", "PO8", "O1", "Oz", "Fp2", "O2", "T8", "FC3"]

# Constants
SCALING_FACTOR = 1e12

# Helper function to compute frequency band powers
def compute_band_powers(epochs, channels, frequency_bands):
    band_powers = []
    for ch in channels:
        epochs_ch = epochs.copy().pick_channels([ch])
        psds, _ = freq_band_power(epochs_ch)
        band_powers.extend([psds[band][0] for band in frequency_bands])
    return np.array(band_powers)

# Compute band powers for placebo and alcohol epochs
band_powers_placebo = compute_band_powers(epochs_placebo, channels, frequency_bands)
band_powers_alcohol = compute_band_powers(epochs_alcohol, channels, frequency_bands)

# Determine colorbar limits
vmin, vmax = min(band_powers_placebo.min(), band_powers_alcohol.min()) * SCALING_FACTOR, \
             max(band_powers_placebo.max(), band_powers_alcohol.max()) * SCALING_FACTOR

# Create subplots
fig, axes = plt.subplots(2, len(frequency_bands), figsize=(16, 4))

# Plot topomaps
epochs_placebo.compute_psd().plot_topomap(vlim=(vmin, vmax), axes=axes[0, :], colorbar=False)
epochs_alcohol.compute_psd().plot_topomap(vlim=(vmin, vmax), axes=axes[1, :], colorbar=False)

plt.tight_layout()
plt.show()

Thank you!