Collection d'ellipses Matplotlib

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.

Skills Graph

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.