Criando Visualizações de Contorno 3D

Introdução

Neste laboratório, aprenderemos como criar um gráfico de contorno 3D usando a biblioteca Matplotlib em Python. Um gráfico de contorno é uma representação gráfica de uma superfície 3D na qual os contornos são plotados em um plano 2D. Gráficos de contorno são úteis para visualizar a variação de uma variável em relação a duas outras variáveis.

Dicas para a VM

Após a inicialização da VM, clique no canto superior esquerdo para mudar para a aba Notebook e acessar o Jupyter Notebook para praticar.

Às vezes, pode ser necessário aguardar alguns segundos para que o Jupyter Notebook termine de carregar. A validação das operações não pode ser automatizada devido a limitações no Jupyter Notebook.

Se você enfrentar problemas durante o aprendizado, sinta-se à vontade para perguntar ao Labby. Forneça feedback após a sessão, e resolveremos o problema prontamente para você.

Importar Bibliotecas

Começaremos importando as bibliotecas necessárias para criar o gráfico de contorno 3D. Usaremos as bibliotecas matplotlib e mpl_toolkits.

import matplotlib.pyplot as plt
from matplotlib import cm
from mpl_toolkits.mplot3d import axes3d

Criar Figura e Subplot

Em seguida, criamos uma figura e um subplot para conter nosso gráfico de contorno 3D.

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

Obter Dados de Teste

Usaremos a função axes3d.get_test_data() para obter alguns dados de teste para plotar.

X, Y, Z = axes3d.get_test_data(0.05)

Criar Gráfico de Contorno

Agora podemos criar um gráfico de contorno 3D dos dados de teste usando a função ax.contour().

ax.contour(X, Y, Z, cmap=cm.coolwarm)

Personalizar o Gráfico

Podemos personalizar o gráfico adicionando rótulos aos eixos e ajustando o ângulo de visualização.

ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')
ax.view_init(elev=30, azim=120)

Mostrar o Gráfico

Finalmente, usamos a função plt.show() para exibir o gráfico de contorno 3D.

plt.show()

Resumo

Neste laboratório, aprendemos como criar um gráfico de contorno 3D usando a biblioteca Matplotlib em Python. Importamos as bibliotecas necessárias, criamos uma figura e um subplot, obtivemos dados de teste, criamos um gráfico de contorno, personalizamos o gráfico e exibimos o gráfico. Gráficos de contorno são úteis para visualizar a variação de uma variável em relação a duas outras variáveis.