Matplotlib - Gráfico de Contorno

Os gráficos de contorno (às vezes chamados de gráficos de nível) são uma forma de mostrar uma superfície tridimensional em um plano bidimensional. Ele representa graficamente duas variáveis ​​preditoras XY no eixo y e uma variável de resposta Z como contornos. Esses contornos são às vezes chamados de fatias z ou valores de iso-resposta.

Um gráfico de contorno é apropriado se você quiser ver como o alor Z muda em função de duas entradas X e Y, de modo que Z = f (X, Y). Uma linha de contorno ou isolinha de uma função de duas variáveis ​​é uma curva ao longo da qual a função tem um valor constante.

As variáveis ​​independentes xey são geralmente restritas a uma grade regular chamada meshgrid. O numpy.meshgrid cria uma grade retangular de uma matriz de valores xe uma matriz de valores y.

A API Matplotlib contém funções contour () e contourf () que desenham linhas de contorno e contornos preenchidos, respectivamente. Ambas as funções precisam de três parâmetros x, y e 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()