Anotar Flecha de Texto con Matplotlib de Python

Introducción

Este tutorial te guiará a través de la creación de un diagrama de dispersión con anotaciones y una flecha de texto utilizando Matplotlib en Python.

Consejos sobre la VM

Una vez finalizada 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 se puede automatizar 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 rápidamente el problema para ti.

Importar bibliotecas y generar datos aleatorios

Primero, necesitamos importar las bibliotecas necesarias y generar algunos datos aleatorios para nuestro diagrama de dispersión.

import matplotlib.pyplot as plt
import numpy as np

## Fijar el estado aleatorio para la reproducibilidad
np.random.seed(19680801)

fig, ax = plt.subplots(figsize=(5, 5))
ax.set_aspect(1)

x1 = -1 + np.random.randn(100)
y1 = -1 + np.random.randn(100)
x2 = 1. + np.random.randn(100)
y2 = 1. + np.random.randn(100)

ax.scatter(x1, y1, color="r")
ax.scatter(x2, y2, color="g")

Agregar anotaciones de texto al gráfico

A continuación, agregaremos anotaciones de texto al gráfico utilizando la función ax.text(). Crearemos dos anotaciones, una para "Muestra A" y otra para "Muestra B".

bbox_props = dict(boxstyle="round", fc="w", ec="0.5", alpha=0.9)
ax.text(-2, -2, "Muestra A", ha="center", va="center", size=20,
        bbox=bbox_props)
ax.text(2, 2, "Muestra B", ha="center", va="center", size=20,
        bbox=bbox_props)

Agregar una flecha de texto para indicar la dirección

Para indicar la dirección de los datos, agregaremos una flecha de texto utilizando la función ax.text() y el parámetro bbox con el boxstyle establecido en "rarrow".

bbox_props = dict(boxstyle="rarrow", fc=(0.8, 0.9, 0.9), ec="b", lw=2)
t = ax.text(0, 0, "Dirección", ha="center", va="center", rotation=45,
            size=15,
            bbox=bbox_props)

bb = t.get_bbox_patch()
bb.set_boxstyle("rarrow", pad=0.6)

Establecer los límites del gráfico y mostrar el gráfico

Finalmente, estableceremos los límites x e y del gráfico y mostraremos el gráfico utilizando las funciones ax.set_xlim(), ax.set_ylim() y plt.show().

ax.set_xlim(-4, 4)
ax.set_ylim(-4, 4)

plt.show()

Resumen

En este tutorial, aprendimos cómo crear un diagrama de dispersión con anotaciones y una flecha de texto utilizando Matplotlib en Python. Utilizamos la función ax.text() para agregar anotaciones y una flecha de texto al gráfico, y las funciones ax.set_xlim(), ax.set_ylim() y plt.show() para establecer los límites del gráfico y mostrar el gráfico.