Use constrained layout in matplotlib visualization (mne-tools#12050)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Eric Larson <[email protected]>

Author: 3 people

README.rst

@@ -96,7 +96,7 @@ The minimum required dependencies to run MNE-Python are:
 - Python >= 3.8
 - NumPy >= 1.21.2
 - SciPy >= 1.7.1
-- Matplotlib >= 3.4.3
+- Matplotlib >= 3.5.0
 - pooch >= 1.5
 - tqdm
 - Jinja2

doc/changes/devel.rst

@@ -37,6 +37,7 @@ Enhancements
 - Add :class:`~mne.time_frequency.EpochsSpectrumArray` and :class:`~mne.time_frequency.SpectrumArray` to support creating power spectra from :class:`NumPy array <numpy.ndarray>` data (:gh:`11803` by `Alex Rockhill`_)
 - Add support for writing forward solutions to HDF5 and convenience function :meth:`mne.Forward.save` (:gh:`12036` by `Eric Larson`_)
 - Refactored internals of :func:`mne.read_annotations` (:gh:`11964` by `Paul Roujansky`_)
+- By default MNE-Python creates matplotlib figures with ``layout='constrained'`` rather than the default ``layout='tight'`` (:gh:`12050` by `Mathieu Scheltienne`_ and `Eric Larson`_)
 - Enhance :func:`~mne.viz.plot_evoked_field` with a GUI that has controls for time, colormap, and contour lines (:gh:`11942` by `Marijn van Vliet`_)
 - Add :class:`mne.viz.ui_events.UIEvent` linking for interactive colorbars, allowing users to link figures and change the colormap and limits interactively. This supports :func:`~mne.viz.plot_evoked_topomap`, :func:`~mne.viz.plot_ica_components`, :func:`~mne.viz.plot_tfr_topomap`, :func:`~mne.viz.plot_projs_topomap`, :meth:`~mne.Evoked.plot_image`, and :meth:`~mne.Epochs.plot_image` (:gh:`12057` by `Santeri Ruuskanen`_)
 

doc/conf.py

@@ -1291,7 +1291,6 @@ def reset_warnings(gallery_conf, fname):
     warnings.filterwarnings("default", module="sphinx")
     # allow these warnings, but don't show them
     for key in (
-        "The module matplotlib.tight_layout is deprecated",  # nilearn
         "invalid version and will not be supported",  # pyxdf
         "distutils Version classes are deprecated",  # seaborn and neo
         "`np.object` is a deprecated alias for the builtin `object`",  # pyxdf

examples/decoding/decoding_rsa_sgskip.py

@@ -150,7 +150,7 @@
 ##############################################################################
 # Plot
 labels = [""] * 5 + ["face"] + [""] * 11 + ["bodypart"] + [""] * 6
-fig, ax = plt.subplots(1)
+fig, ax = plt.subplots(1, layout="constrained")
 im = ax.matshow(confusion, cmap="RdBu_r", clim=[0.3, 0.7])
 ax.set_yticks(range(len(classes)))
 ax.set_yticklabels(labels)
@@ -159,14 +159,13 @@
 ax.axhline(11.5, color="k")
 ax.axvline(11.5, color="k")
 plt.colorbar(im)
-plt.tight_layout()
 plt.show()
 
 ##############################################################################
 # Confusion matrix related to mental representations have been historically
 # summarized with dimensionality reduction using multi-dimensional scaling [1].
 # See how the face samples cluster together.
-fig, ax = plt.subplots(1)
+fig, ax = plt.subplots(1, layout="constrained")
 mds = MDS(2, random_state=0, dissimilarity="precomputed")
 chance = 0.5
 summary = mds.fit_transform(chance - confusion)
@@ -186,7 +185,6 @@
     )
 ax.axis("off")
 ax.legend(loc="lower right", scatterpoints=1, ncol=2)
-plt.tight_layout()
 plt.show()
 
 ##############################################################################

examples/decoding/decoding_spoc_CMC.py

@@ -68,15 +68,14 @@
 y_preds = cross_val_predict(clf, X, y, cv=cv)
 
 # Plot the True EMG power and the EMG power predicted from MEG data
-fig, ax = plt.subplots(1, 1, figsize=[10, 4])
+fig, ax = plt.subplots(1, 1, figsize=[10, 4], layout="constrained")
 times = raw.times[meg_epochs.events[:, 0] - raw.first_samp]
 ax.plot(times, y_preds, color="b", label="Predicted EMG")
 ax.plot(times, y, color="r", label="True EMG")
 ax.set_xlabel("Time (s)")
 ax.set_ylabel("EMG Power")
 ax.set_title("SPoC MEG Predictions")
 plt.legend()
-mne.viz.tight_layout()
 plt.show()
 
 ##############################################################################

examples/decoding/decoding_time_generalization_conditions.py

@@ -88,7 +88,7 @@
 
 # %%
 # Plot
-fig, ax = plt.subplots(constrained_layout=True)
+fig, ax = plt.subplots(layout="constrained")
 im = ax.matshow(
     scores,
     vmin=0,

examples/decoding/decoding_xdawn_eeg.py

@@ -99,22 +99,24 @@
 cm_normalized = cm.astype(float) / cm.sum(axis=1)[:, np.newaxis]
 
 # Plot confusion matrix
-fig, ax = plt.subplots(1)
+fig, ax = plt.subplots(1, layout="constrained")
 im = ax.imshow(cm_normalized, interpolation="nearest", cmap=plt.cm.Blues)
 ax.set(title="Normalized Confusion matrix")
 fig.colorbar(im)
 tick_marks = np.arange(len(target_names))
 plt.xticks(tick_marks, target_names, rotation=45)
 plt.yticks(tick_marks, target_names)
-fig.tight_layout()
 ax.set(ylabel="True label", xlabel="Predicted label")
 
 # %%
 # The ``patterns_`` attribute of a fitted Xdawn instance (here from the last
 # cross-validation fold) can be used for visualization.
 
 fig, axes = plt.subplots(
-    nrows=len(event_id), ncols=n_filter, figsize=(n_filter, len(event_id) * 2)
+    nrows=len(event_id),
+    ncols=n_filter,
+    figsize=(n_filter, len(event_id) * 2),
+    layout="constrained",
 )
 fitted_xdawn = clf.steps[0][1]
 info = create_info(epochs.ch_names, 1, epochs.get_channel_types())
@@ -131,7 +133,6 @@
         show=False,
     )
     axes[ii, 0].set(ylabel=cur_class)
-fig.tight_layout(h_pad=1.0, w_pad=1.0, pad=0.1)
 
 # %%
 # References

examples/decoding/receptive_field_mtrf.py

@@ -67,12 +67,11 @@
 n_channels = len(raw.ch_names)
 
 # Plot a sample of brain and stimulus activity
-fig, ax = plt.subplots()
+fig, ax = plt.subplots(layout="constrained")
 lns = ax.plot(scale(raw[:, :800][0].T), color="k", alpha=0.1)
 ln1 = ax.plot(scale(speech[0, :800]), color="r", lw=2)
 ax.legend([lns[0], ln1[0]], ["EEG", "Speech Envelope"], frameon=False)
 ax.set(title="Sample activity", xlabel="Time (s)")
-mne.viz.tight_layout()
 
 # %%
 # Create and fit a receptive field model
@@ -117,12 +116,11 @@
 mean_scores = scores.mean(axis=0)
 
 # Plot mean prediction scores across all channels
-fig, ax = plt.subplots()
+fig, ax = plt.subplots(layout="constrained")
 ix_chs = np.arange(n_channels)
 ax.plot(ix_chs, mean_scores)
 ax.axhline(0, ls="--", color="r")
 ax.set(title="Mean prediction score", xlabel="Channel", ylabel="Score ($r$)")
-mne.viz.tight_layout()
 
 # %%
 # Investigate model coefficients
@@ -134,7 +132,7 @@
 
 # Print mean coefficients across all time delays / channels (see Fig 1)
 time_plot = 0.180  # For highlighting a specific time.
-fig, ax = plt.subplots(figsize=(4, 8))
+fig, ax = plt.subplots(figsize=(4, 8), layout="constrained")
 max_coef = mean_coefs.max()
 ax.pcolormesh(
     times,
@@ -155,16 +153,14 @@
     xticks=np.arange(tmin, tmax + 0.2, 0.2),
 )
 plt.setp(ax.get_xticklabels(), rotation=45)
-mne.viz.tight_layout()
 
 # Make a topographic map of coefficients for a given delay (see Fig 2C)
 ix_plot = np.argmin(np.abs(time_plot - times))
-fig, ax = plt.subplots()
+fig, ax = plt.subplots(layout="constrained")
 mne.viz.plot_topomap(
     mean_coefs[:, ix_plot], pos=info, axes=ax, show=False, vlim=(-max_coef, max_coef)
 )
 ax.set(title="Topomap of model coefficients\nfor delay %s" % time_plot)
-mne.viz.tight_layout()
 
 # %%
 # Create and fit a stimulus reconstruction model
@@ -240,15 +236,14 @@
 
 y_pred = sr.predict(Y[test])
 time = np.linspace(0, 2.0, 5 * int(sfreq))
-fig, ax = plt.subplots(figsize=(8, 4))
+fig, ax = plt.subplots(figsize=(8, 4), layout="constrained")
 ax.plot(
     time, speech[test][sr.valid_samples_][: int(5 * sfreq)], color="grey", lw=2, ls="--"
 )
 ax.plot(time, y_pred[sr.valid_samples_][: int(5 * sfreq)], color="r", lw=2)
 ax.legend([lns[0], ln1[0]], ["Envelope", "Reconstruction"], frameon=False)
 ax.set(title="Stimulus reconstruction")
 ax.set_xlabel("Time (s)")
-mne.viz.tight_layout()
 
 # %%
 # Investigate model coefficients
@@ -292,7 +287,6 @@
     title="Inverse-transformed coefficients\nbetween delays %s and %s"
     % (time_plot[0], time_plot[1])
 )
-mne.viz.tight_layout()
 
 # %%
 # References

examples/inverse/label_source_activations.py

@@ -62,7 +62,7 @@
 # View source activations
 # -----------------------
 
-fig, ax = plt.subplots(1)
+fig, ax = plt.subplots(1, layout="constrained")
 t = 1e3 * stc_label.times
 ax.plot(t, stc_label.data.T, "k", linewidth=0.5, alpha=0.5)
 pe = [
@@ -81,7 +81,6 @@
     xlim=xlim,
     ylim=ylim,
 )
-mne.viz.tight_layout()
 
 # %%
 # Using vector solutions
@@ -92,7 +91,7 @@
 pick_ori = "vector"
 stc_vec = apply_inverse(evoked, inverse_operator, lambda2, method, pick_ori=pick_ori)
 data = stc_vec.extract_label_time_course(label, src)
-fig, ax = plt.subplots(1)
+fig, ax = plt.subplots(1, layout="constrained")
 stc_vec_label = stc_vec.in_label(label)
 colors = ["#EE6677", "#228833", "#4477AA"]
 for ii, name in enumerate("XYZ"):
@@ -117,4 +116,3 @@
     xlim=xlim,
     ylim=ylim,
 )
-mne.viz.tight_layout()

examples/inverse/mixed_source_space_inverse.py

@@ -194,9 +194,8 @@
 )
 
 # plot the times series of 2 labels
-fig, axes = plt.subplots(1)
+fig, axes = plt.subplots(1, layout="constrained")
 axes.plot(1e3 * stc.times, label_ts[0][0, :], "k", label="bankssts-lh")
 axes.plot(1e3 * stc.times, label_ts[0][-1, :].T, "r", label="Brain-stem")
 axes.set(xlabel="Time (ms)", ylabel="MNE current (nAm)")
 axes.legend()
-mne.viz.tight_layout()