Tracer une ligne droite avec matplotlib

Pour tracer une simple ligne droite avec matplotlib il existe la fonction plot du module pyplot dont voici un exemple simple d'utilisation:

import matplotlib.pyplot as plt
import numpy as np



plt.plot([0.0, 10.0], [0.0, 10.0], 'r-', lw=2) # Red straight line
plt.plot([2.0, 6.0], [0.0, 10.0], 'b--', lw=2) # Blue dashed straight line

plt.savefig('StraightLine.png')
plt.show()

Le plus simple pour tracer une ligne droite entre les points (x1,y1) et (x2,y2) est d'écrire la commande suivante: plt.plot([x1, x2], [y1, y2]). Vous pouvez alors ajouter des arguments supplémentaires comme la couleur c='r', l'épaisseur de la ligne lw=2, etc (voir matplotlib pyplot pour avoir l'ensembles des arguments possibles). Il est possible aussi d'utiliser plot en ne donnant que la liste des y: plt.plot([y1, y2]) mais dans ce cas la fonction suppose que la liste des abscisses correspond aux indices de la liste des ordonnées donc ici x1=0 et x2=1. Exemple:

plt.plot([0.0, 4.0])

Exemple légèrement plus complexe utilisant une fonction pour définir la liste des y:

 import matplotlib.pyplot as plt
 import numpy as np

 def f(x):
      return 0.4 * x + 2

plt.plot([f(x) for x in range(11)])

 plt.show()

Recherches associées