Tracer un graphique à l'aide de python

Matplotlib est un module de Python permettant de tracer des graphiques. Dans ce module, on va utiliser le sous module pyplot que l'on va renommer plt. Cela se fait de cette façon :

In [1]:
from matplotlib import pyplot as plt

Imaginons maintenant que nous souhaitions tracer la fonction f(x)=2x²-x +2 pour x € [-4;4]. pyplot ne sait pas réellement tracer une fonction : il peut placer des listes de points sur un graphique et les relier à l'aide d'une droite. On va donc créer 2 listes, liste_x et liste_y dans lesquelles on va placer les valeurs des abscisses et des ordonnées. Pour calculer ces points, on va créer une fonction f(x) qui va se charger de calculer les valeurs de y pour chaque x qu'on va lui donner et une fonction generation(a,b,nb_pts) qui à partir de l'intervalle [a;b] donné et du nombre de points nb_pts indiqués va se charger de définir tous les x à calculer et créer les listes liste_x et liste_y. Une fois ces listes créées, on les fera tracer à pyplot

In [5]:
def f(x):
    return 2*x**2-x+2

def generation(a,b,nb_pts):
    liste_x,liste_y=[],[]
    for i in range(nb_pts):
        x=a+i*(b-a)/nb_pts
        liste_x.append(x)
        liste_y.append(f(x))
    plt.plot(liste_x,liste_y)
    plt.show()

generation(-4,4,100)

L'avantage de pyplot est qu'il calcule directement l'échelle la plus appropriée pour afficher le graphique.