Visualisation de diagramme en tiges avec Python

Introduction

Le diagramme en tiges est un type de graphique utilisé dans la visualisation de données. Il est utilisé pour tracer des lignes verticales d'une ligne de base jusqu'à la coordonnée y et placer un marqueur à l'extrémité. Dans ce laboratoire, nous allons apprendre à utiliser la fonction de diagramme en tiges dans la bibliothèque Matplotlib de Python.

Conseils sur la VM

Une fois le démarrage de la VM terminé, cliquez dans le coin supérieur gauche pour basculer vers l'onglet Carnet de notes 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 limites 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ésoudrons rapidement le problème pour vous.

Importation des bibliothèques

Avant de commencer, nous devons importer les bibliothèques nécessaires. Dans ce cas, nous allons utiliser les bibliothèques Matplotlib et Numpy.

import matplotlib.pyplot as plt
import numpy as np

Générer des données

Ensuite, nous devons générer des données à utiliser dans notre diagramme en tiges. Nous allons créer deux tableaux à l'aide de la bibliothèque Numpy.

x = np.linspace(0.1, 2 * np.pi, 41)
y = np.exp(np.sin(x))

Créer un diagramme en tiges de base

Nous pouvons maintenant créer un diagramme en tiges de base à l'aide de la fonction stem de la bibliothèque Matplotlib.

plt.stem(x, y)
plt.show()

Cela générera un graphique avec des lignes verticales allant de la ligne de base jusqu'à la coordonnée y et des marqueurs à l'extrémité.

Personnaliser le graphique

Nous pouvons personnaliser le graphique en ajustant la ligne de base à l'aide du paramètre bottom. Nous pouvons également ajuster les propriétés de format du graphique à l'aide des paramètres linefmt, markerfmt et basefmt.

markerline, stemlines, baseline = plt.stem(
    x, y, linefmt='grey', markerfmt='D', bottom=1.1)
markerline.set_markerfacecolor('none')
plt.show()

Cela générera un graphique avec un format de ligne gris et des marqueurs en forme de losange. La ligne de base a également été ajustée à 1,1.

Sommaire

Dans ce laboratoire, nous avons appris à utiliser la fonction de diagramme en tiges de la bibliothèque Matplotlib de Python. Nous avons tout d'abord importé les bibliothèques nécessaires, généré des données et créé un diagramme en tiges de base. Nous avons ensuite personnalisé le graphique en ajustant la ligne de base et les propriétés de format. En suivant ces étapes, nous pouvons créer des diagrammes en tiges informatifs et visuellement attrayants pour nos données.