Matplotlib - Tracé de contour

Les tracés de contour (parfois appelés tracés de niveau) sont un moyen de montrer une surface tridimensionnelle sur un plan bidimensionnel. Il trace graphiquement deux variables prédictives XY sur l'axe des y et une variable de réponse Z sous forme de contours. Ces contours sont parfois appelés les z-slices ou les valeurs d'iso-réponse.

Un tracé de contour est approprié si vous voulez voir comment la aleur Z change en fonction de deux entrées X et Y, telles que Z = f (X, Y). Une ligne de contour ou isoligne d'une fonction de deux variables est une courbe le long de laquelle la fonction a une valeur constante.

Les variables indépendantes x et y sont généralement limitées à une grille régulière appelée meshgrid. Le numpy.meshgrid crée une grille rectangulaire à partir d'un tableau de valeurs x et d'un tableau de valeurs y.

L'API Matplotlib contient des fonctions contour () et contourf () qui dessinent respectivement des lignes de contour et des contours remplis. Les deux fonctions nécessitent trois paramètres x, y et z.

import numpy as np
import matplotlib.pyplot as plt
xlist = np.linspace(-3.0, 3.0, 100)
ylist = np.linspace(-3.0, 3.0, 100)
X, Y = np.meshgrid(xlist, ylist)
Z = np.sqrt(X**2 + Y**2)
fig,ax=plt.subplots(1,1)
cp = ax.contourf(X, Y, Z)
fig.colorbar(cp) # Add a colorbar to a plot
ax.set_title('Filled Contours Plot')
#ax.set_xlabel('x (cm)')
ax.set_ylabel('y (cm)')
plt.show()