Annoter une flèche de texte

Introduction

Ce tutoriel vous guidera tout au long de la création d'un graphique en dispersion annoté avec une flèche de texte à l'aide de Matplotlib en Python.

Conseils sur la machine virtuelle

Une fois le démarrage de la machine virtuelle terminé, cliquez sur le coin supérieur gauche pour basculer vers l'onglet Carnet d'adresses pour accéder au carnet Jupyter pour pratiquer.

Parfois, vous devrez peut-être attendre quelques secondes pour que le carnet Jupyter ait fini de charger. La validation des opérations ne peut pas être automatisée en raison des limitations du carnet Jupyter.

Si vous rencontrez des problèmes pendant l'apprentissage, n'hésitez pas à demander à Labby. Donnez votre feedback après la session, et nous réglerons rapidement le problème pour vous.

Importation des bibliothèques et génération de données aléatoires

Tout d'abord, nous devons importer les bibliothèques nécessaires et générer quelques données aléatoires pour notre graphique en dispersion.

import matplotlib.pyplot as plt
import numpy as np

## Fixing random state for reproducibility
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")

Ajout d'annotations de texte au graphique

Ensuite, nous ajouterons des annotations de texte au graphique en utilisant la fonction ax.text(). Nous créerons deux annotations, l'une pour "Échantillon A" et l'autre pour "Échantillon B".

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

Ajouter une flèche de texte pour indiquer la direction

Pour indiquer la direction des données, nous ajouterons une flèche de texte en utilisant la fonction ax.text() et le paramètre bbox avec le boxstyle défini sur "rarrow".

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

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

Définir les limites du graphique et afficher le graphique

Enfin, nous définirons les limites x et y du graphique et afficherons le graphique en utilisant les fonctions ax.set_xlim(), ax.set_ylim() et plt.show().

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

plt.show()

Sommaire

Dans ce tutoriel, nous avons appris à créer un graphique en dispersion annoté avec une flèche de texte à l'aide de Matplotlib en Python. Nous avons utilisé la fonction ax.text() pour ajouter des annotations et une flèche de texte au graphique, et les fonctions ax.set_xlim(), ax.set_ylim() et plt.show() pour définir les limites du graphique et afficher le graphique.