Como criar um gráfico de contorno no matplotlib

Um gráfico de contorno é um tipo de gráfico que nos permite visualizar dados tridimensionais em duas dimensões usando contornos.

Você pode criar um gráfico de contorno no Matplotlib usando as duas funções a seguir:

• matplotlib.pyplot.contour() – Cria gráficos de contorno.
• matplotlib.pyplot.contourf() – Cria gráficos de contorno preenchidos.

Os exemplos a seguir mostram como usar essas duas funções na prática.

Exemplo 1: Gráfico de contorno em Matplotlib

Suponha que temos os seguintes dados em Python:

 import numpy as np

x = np.linspace(0, 5, 50)
y = np.linspace(0, 5, 40)

X, Y = np.meshgrid(x, y)
Z = np.sin(X*2+Y)*3 + np.cos(Y+5)

Podemos usar o seguinte código para criar um gráfico de contorno para os dados:

 import matplotlib.pyplot as plt

plt. outline (X, Y, Z, colors=' black ')

Quando apenas uma cor é usada para o gráfico, as linhas tracejadas representam valores negativos e as linhas sólidas representam valores positivos.

Uma alternativa é especificar um mapa de cores usando o argumento cmap . Também podemos especificar mais linhas para usar no gráfico com o argumento de níveis :

 plt. contour (X, Y, Z, levels= 30 , cmap=' Reds ') 

Optamos por usar o cmap ‘Reds’, mas você pode encontrar uma lista completa de opções de paleta de cores na página de documentação do Matplotlib .

Exemplo 2: Gráfico de contorno preenchido em Matplotlib

Um caminho de contorno preenchido é semelhante a um caminho de contorno, exceto que os espaços entre as linhas são preenchidos.

O código a seguir mostra como usar a função contornof() para criar um gráfico de contorno preenchido para os mesmos dados usados no exemplo anterior:

 plt. contourf (X, Y, Z, cmap=' Reds ') 

Também podemos usar a função colorbar() para adicionar uma barra de cores rotulada ao lado do gráfico:

 plt. contourf (X, Y, Z, cmap=' Reds ')
plt. colorbar () 

Você pode encontrar mais tutoriais do Matplotlib aqui .