Introduction
Matplotlib est une puissante bibliothèque de visualisation de données en Python. Dans ce laboratoire, nous allons explorer l'utilisation de EllipseCollection pour tracer une collection d'ellipses.
Conseils sur la machine virtuelle
Une fois le démarrage de la machine virtuelle terminé, cliquez dans le coin supérieur gauche pour basculer vers l'onglet Carnet de notes pour accéder au carnet Jupyter pour la pratique.
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 des commentaires après la session, et nous résoudrons rapidement le problème pour vous.
Importation des bibliothèques nécessaires
Nous allons commencer par importer les bibliothèques nécessaires.
import matplotlib.pyplot as plt
import numpy as np
from matplotlib.collections import EllipseCollection
Création de données pour les ellipses
Nous créons les données pour nos ellipses sous forme de tableaux de coordonnées x, de coordonnées y, de largeur, de hauteur et d'angle.
x = np.arange(10)
y = np.arange(15)
X, Y = np.meshgrid(x, y)
XY = np.column_stack((X.ravel(), Y.ravel()))
ww = X / 10.0
hh = Y / 15.0
aa = X * 9
Création d'une collection d'ellipses
Nous créons une EllipseCollection avec les données ci-dessus et spécifions que les unités sont 'x' et les décalages sont XY.
fig, ax = plt.subplots()
ec = EllipseCollection(ww, hh, aa, units='x', offsets=XY,
offset_transform=ax.transData)
Définir la couleur des ellipses
Nous définissons la couleur de chaque ellipse dans la EllipseCollection en fonction de la somme de ses coordonnées x et y.
ec.set_array((X + Y).ravel())
Ajouter la collection au graphique
Nous ajoutons la EllipseCollection au graphique.
ax.add_collection(ec)
ax.autoscale_view()
ax.set_xlabel('X')
ax.set_ylabel('y')
cbar = plt.colorbar(ec)
cbar.set_label('X+Y')
plt.show()
Sommaire
Dans ce laboratoire, nous avons appris à utiliser EllipseCollection pour tracer une collection d'ellipses dans Matplotlib. Nous avons également appris à définir la couleur de chaque ellipse en fonction de ses coordonnées x et y.