Creando subdiagramas con Matplotlib en Python

Introducción

Matplotlib es una biblioteca de Python que se utiliza para crear visualizaciones como diagramas de líneas, diagramas de dispersión, diagramas de barras y subdiagramas, entre otros. En este laboratorio, aprenderá a crear una figura con dos subdiagramas utilizando .pyplot.subplot.

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 su retroalimentación después de la sesión y lo resolveremos rápidamente para usted.

Importar las bibliotecas necesarias

Para este tutorial, usaremos el módulo pyplot de la biblioteca Matplotlib y la biblioteca numpy.

import matplotlib.pyplot as plt
import numpy as np

Definir los datos

Definiremos dos conjuntos de datos que usaremos para crear nuestros subdiagramas.

def f(t):
    return np.exp(-t) * np.cos(2*np.pi*t)

t1 = np.arange(0.0, 5.0, 0.1)
t2 = np.arange(0.0, 5.0, 0.02)

Crear los subdiagramas

Crearemos una figura con dos subdiagramas utilizando .pyplot.subplot.

plt.figure()

plt.subplot(211)
plt.plot(t1, f(t1), color='tab:blue', marker='o')
plt.plot(t2, f(t2), color='black')

plt.subplot(212)
plt.plot(t2, np.cos(2*np.pi*t2), color='tab:orange', linestyle='--')

plt.show()

La función subplot() toma tres argumentos: el número de filas, el número de columnas y el índice del diagrama actual. El índice comienza en 1 en la esquina superior izquierda y aumenta por filas. En este ejemplo, creamos una figura con dos subdiagramas: uno arriba y uno abajo.

En el primer subdiagrama, graficamos t1 contra f(t1) y t2 contra f(t2). Establecemos el color de la primera gráfica en azul y agregamos marcadores circulares a cada punto de datos. Establecemos el color de la segunda gráfica en negro.

En el segundo subdiagrama, graficamos t2 contra la función coseno de 2*np.pi*t2. Establecemos el color de la gráfica en naranja y el estilo de línea en discontinuo.

Referencias

En este ejemplo se muestra el uso de las siguientes funciones, métodos, clases y módulos:

• matplotlib.pyplot.figure
• matplotlib.pyplot.subplot

Resumen

En este laboratorio, aprendimos cómo crear una figura con dos subdiagramas utilizando .pyplot.subplot. Definimos dos conjuntos de datos y los graficamos en dos subdiagramas usando diferentes colores y estilos de línea. También proporcionamos referencias a las funciones y módulos utilizados en este ejemplo.