Diagrama de dispersión animado con Matplotlib

Introducción

Esta práctica está diseñada para enseñarte a crear un diagrama de dispersión animado utilizando la biblioteca Matplotlib de Python. Cubriremos todo desde la configuración del diagrama hasta la guardado de la animación como un archivo GIF. Al final de esta práctica, tendrás un diagrama de dispersión animado funcional que puedes utilizar para visualizar tus datos.

Consejos sobre la VM

Una vez que se haya completado la inicialización de la VM, haz 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 tengas que esperar unos segundos a que Jupyter Notebook termine de cargarse. La validación de operaciones no puede automatizarse debido a las limitaciones de Jupyter Notebook.

Si tienes problemas durante el aprendizaje, no dudes en preguntar a Labby. Proporciona retroalimentación después de la sesión y resolveremos el problema para ti de inmediato.

Configuración del diagrama

El primer paso para crear un diagrama de dispersión animado es configurar el diagrama en sí mismo. Esto implica importar las bibliotecas necesarias y crear un objeto de figura y ejes.

import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots()

Definiendo los datos

A continuación, necesitamos definir los datos que utilizaremos para nuestro diagrama de dispersión. En este ejemplo, usaremos una simple matriz de números que van desde 0 hasta 10.

x = np.linspace(0, 10)

Creando el diagrama de dispersión

Ahora que tenemos nuestros datos, podemos crear el diagrama de dispersión. Hacemos esto llamando a la función scatter en nuestro objeto de ejes y pasando nuestros datos de x.

scat = ax.scatter(1, 0)

Creando la animación

El último paso es crear la animación. Hacemos esto utilizando la función FuncAnimation del módulo animation. Esta función toma varios argumentos, incluyendo el objeto de figura, la función que actualizará el diagrama y el número de fotogramas a utilizar.

def animate(i):
    scat.set_offsets((x[i], 0))
    return scat,

ani = animation.FuncAnimation(fig, animate, repeat=True,
                                    frames=len(x) - 1, interval=50)

Mostrando el diagrama

Ahora podemos mostrar el diagrama llamando a la función show del módulo pyplot.

plt.show()

Resumen

En este laboratorio, aprendimos cómo crear un diagrama de dispersión animado utilizando la biblioteca Matplotlib de Python. Cubrimos todo desde la configuración del diagrama hasta guardar la animación como un GIF. Con este conocimiento, ahora puedes crear tus propios diagramas de dispersión animados para visualizar tus datos.