Création de sous-graphiques avec Matplotlib en Python

Introduction

Matplotlib est une bibliothèque Python utilisée pour créer des visualisations telles que des graphiques linéaires, des graphiques de dispersion, des graphiques en barres et des sous-graphiques, entre autres. Dans ce laboratoire, vous allez apprendre à créer une figure avec deux sous-graphiques à l'aide de .pyplot.subplot.

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 d'étude pour accéder à Jupyter Notebook pour la pratique.

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

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.

Importez les bibliothèques nécessaires

Pour ce tutoriel, nous allons utiliser le module pyplot de la bibliothèque Matplotlib et la bibliothèque numpy.

import matplotlib.pyplot as plt
import numpy as np

Définissez les données

Nous allons définir deux ensembles de données que nous utiliserons pour créer nos sous-graphiques.

def f(t):
    return np.exp(-t) * np.cos(2*np.pi*t)

t1 = np.arange(0.0, 5.0, 0.1)
t2 = np.arange(0.0, 5.0, 0.02)

Créez les sous-graphiques

Nous allons créer une figure avec deux sous-graphiques à l'aide de .pyplot.subplot.

plt.figure()

plt.subplot(211)
plt.plot(t1, f(t1), color='tab:blue', marker='o')
plt.plot(t2, f(t2), color='black')

plt.subplot(212)
plt.plot(t2, np.cos(2*np.pi*t2), color='tab:orange', linestyle='--')

plt.show()

La fonction subplot() prend trois arguments : le nombre de lignes, le nombre de colonnes et l'index du tracé actuel. L'index commence à 1 dans le coin supérieur gauche et augmente ligne par ligne. Dans cet exemple, nous créons une figure avec deux sous-graphiques : l'un en haut et l'autre en bas.

Dans le premier sous-graphique, nous traçons t1 en fonction de f(t1) et t2 en fonction de f(t2). Nous définissons la couleur du premier tracé en bleu et ajoutons des marqueurs circulaires à chaque point de données. Nous définissons la couleur du second tracé en noir.

Dans le second sous-graphique, nous traçons t2 en fonction de la fonction cosinus de 2*np.pi*t2. Nous définissons la couleur du tracé en orange et le style de ligne en tirets.

Ressources

L'utilisation des fonctions, méthodes, classes et modules suivants est illustrée dans cet exemple :

• matplotlib.pyplot.figure
• matplotlib.pyplot.subplot

Résumé

Dans ce laboratoire, nous avons appris à créer une figure avec deux sous-graphiques à l'aide de .pyplot.subplot. Nous avons défini deux ensembles de données et les avons tracées dans deux sous-graphiques en utilisant différentes couleurs et styles de ligne. Nous avons également fourni des références aux fonctions et aux modules utilisés dans cet exemple.