Tutorial de generación de subtramas en Matplotlib

Introducción

En este tutorial, aprenderá a usar la función subplot2grid de la biblioteca Matplotlib de Python para generar subtramas.

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 las 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 las bibliotecas necesarias

Antes de comenzar, debemos importar la biblioteca Matplotlib con el siguiente código:

import matplotlib.pyplot as plt

Crear un objeto de figura

Para crear un objeto de figura, use el siguiente código:

fig = plt.figure()

Definir subtramas usando subplot2grid

Para definir subtramas usando subplot2grid, primero debemos especificar el tamaño de la cuadrícula usando una tupla con el número deseado de filas y columnas. También debemos especificar la ubicación de la subtrama dentro de la cuadrícula usando otra tupla.

Por ejemplo, para crear una cuadrícula 3x3 con una subtrama que abarque toda la primera fila y las tres columnas, usamos el siguiente código:

ax1 = plt.subplot2grid((3, 3), (0, 0), colspan=3)

Para crear una subtrama que abarque la segunda fila y las dos primeras columnas, usamos:

ax2 = plt.subplot2grid((3, 3), (1, 0), colspan=2)

Para crear una subtrama que abarque las dos últimas filas y la última columna, usamos:

ax3 = plt.subplot2grid((3, 3), (1, 2), rowspan=2)

Para crear una subtrama en la última fila y la primera columna, usamos:

ax4 = plt.subplot2grid((3, 3), (2, 0))

Para crear una subtrama en la última fila y la segunda columna, usamos:

ax5 = plt.subplot2grid((3, 3), (2, 1))

Anotar los ejes

Para anotar los ejes, podemos recorrer los ejes de la figura y agregar texto usando la función text y la función tick_params para eliminar las etiquetas de los ticks.

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)

Mostrar la gráfica

Para mostrar la gráfica, use el siguiente código:

plt.show()

Resumen

En este tutorial, aprendiste cómo usar la función subplot2grid en Matplotlib para generar subtramas. También aprendiste cómo crear un objeto de figura, definir subtramas dentro de una cuadrícula, anotar los ejes y mostrar la gráfica.