Personalización de subtramas con GridSpec

Introducción

En este tutorial, aprenderemos a usar GridSpec para generar subtramas, controlar los tamaños relativos de las subtramas con width_ratios y height_ratios, y el espaciado alrededor y entre las subtramas usando subplot params (left, right, bottom, top, wspace y hspace).

Consejos sobre la VM

Una vez finalizada la inicialización de la VM, haga clic en la esquina superior izquierda para cambiar a la pestaña Cuaderno y acceder a Jupyter Notebook para practicar.

A veces, es posible que tenga que esperar unos segundos a que Jupyter Notebook termine de cargarse. La validación de operaciones no se puede automatizar debido a las limitaciones de Jupyter Notebook.

Si tiene problemas durante el aprendizaje, no dude en preguntar a Labby. Deje sus comentarios después de la sesión y lo resolveremos rápidamente para usted.

Importar bibliotecas

Comenzamos importando las bibliotecas necesarias, que son matplotlib.pyplot y GridSpec.

import matplotlib.pyplot as plt
from matplotlib.gridspec import GridSpec

Generar subtramas con GridSpec

En este paso, usaremos GridSpec para generar subtramas. Crearemos una figura con 2 filas y 2 columnas. También especificaremos width_ratios y height_ratios para controlar los tamaños relativos de las subtramas.

fig = plt.figure()
gs = GridSpec(2, 2, width_ratios=[1, 2], height_ratios=[4, 1])
ax1 = fig.add_subplot(gs[0])
ax2 = fig.add_subplot(gs[1])
ax3 = fig.add_subplot(gs[2])
ax4 = fig.add_subplot(gs[3])

Controlar el espaciado alrededor y entre las subtramas

En este paso, usaremos GridSpec para controlar el espaciado alrededor y entre las subtramas. Crearemos una figura con 2 gridspecs, cada uno con 3 filas y 3 columnas. Especificaremos los parámetros left, right, bottom, top, wspace y hspace para controlar el espaciado.

fig = plt.figure()
gs1 = GridSpec(3, 3, left=0.05, right=0.48, wspace=0.05)
ax1 = fig.add_subplot(gs1[:-1, :])
ax2 = fig.add_subplot(gs1[-1, :-1])
ax3 = fig.add_subplot(gs1[-1, -1])

gs2 = GridSpec(3, 3, left=0.55, right=0.98, hspace=0.05)
ax4 = fig.add_subplot(gs2[:, :-1])
ax5 = fig.add_subplot(gs2[:-1, -1])
ax6 = fig.add_subplot(gs2[-1, -1])

Anotar los ejes

En este paso, anotaremos los ejes con sus respectivos números de subtrama.

def annotate_axes(fig):
    for i, ax in enumerate(fig.axes):
        ax.text(0.5, 0.5, "ax%d" % (i+1), va="center", ha="center")
        ax.tick_params(labelbottom=False, labelleft=False)

annotate_axes(fig)

Mostrar los gráficos

En este paso, mostraremos los gráficos.

plt.show()

Resumen

En este tutorial, aprendimos cómo usar GridSpec para generar subtramas y controlar el espaciado alrededor y entre las subtramas. También aprendimos cómo anotar los ejes con sus respectivos números de subtrama. Estas habilidades serán útiles para crear gráficos y visualizaciones complejas.