Gráficos de Contorno do Matplotlib

Os contornos (às vezes chamados de 'gráficos horizontais') são uma maneira de mostrar superfícies tridimensionais em um plano bidimensional. Ele desenha as variáveis preditoras X Y no eixo y e a variável de resposta do contorno Z. Esses contornos são às vezes chamados de fatias Z ou valores de resposta equivalentes.

Se você quiser ver como Z varia com os dois inputs X e Y, os contornos são muito úteis, por exemplo, Z = f(X, Y). As curvas de nível ou curvas de nível de duas variáveis são curvas que têm valores constantes.

Os variáveis independentes x e y geralmente são limitadas a uma grade regular chamada meshgrid. O numpy.meshgrid cria uma grade retangular usando arrays de valores x e y.

A API do Matplotlib inclui as funções contour() e contourf() para desenhar contornos e preencher contornos separadamente. Ambas as funções precisam de três parâmetros: x, y e z.

# Filename: example.py
# Copyright:　2020 By w3codebox
# Author by: pt.oldtoolbag.com
# Data:　2020-08-08
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) # Adicionar uma barra de cores a um gráfico
　ax.set_title('Gráfico de Contorno do Matplotlib')
　#ax.set_xlabel('x (cm)')
　ax.set_ylabel('y (cm)')
　plt.show()

Execute o código de exemplo acima para obter os seguintes resultados -