Criar Gráficos de Contorno 3D com Matplotlib

Beginner

This tutorial is from open-source community. Access the source code

Introdução

Neste laboratório, você aprenderá como criar um gráfico de contorno 3D usando Matplotlib em Python. Um gráfico de contorno é uma representação gráfica da relação entre três variáveis. Ele é usado para exibir a relação entre duas variáveis nos eixos x e y e a terceira variável no eixo z. Gráficos de contorno são amplamente utilizados em campos científicos e de engenharia para exibir dados em um espaço 3D.

Dicas para a VM

Após a inicialização da VM, clique no canto superior esquerdo para mudar para a aba Notebook para 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 as Bibliotecas Necessárias

Antes de começarmos, precisamos importar as bibliotecas necessárias. Usaremos Matplotlib e Axes3D de mpl_toolkits.mplot3d.

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

Criar os Objetos Figura e Eixos

Agora criaremos os objetos figura e eixos usando o método add_subplot(). Definiremos o parâmetro projection como '3d' para criar um gráfico 3D.

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

Gerar os Dados

Agora geraremos os dados a serem usados no gráfico de contorno 3D. Usaremos o método axes3d.get_test_data() para gerar os dados. Este método gera dados de teste para um gráfico 3D.

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

Criar o Gráfico de Contorno

Agora criaremos o gráfico de contorno usando o método contourf(). Este método cria contornos preenchidos. Definiremos o parâmetro cmap como cm.coolwarm para usar o mapa de cores cool-warm.

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

Exibir o Gráfico

Agora exibiremos o gráfico usando o método show().

plt.show()

Resumo

Neste laboratório, você aprendeu como criar um gráfico de contorno 3D usando Matplotlib em Python. Você aprendeu como importar as bibliotecas necessárias, criar os objetos de figura e eixos, gerar os dados, criar o gráfico de contorno e exibir o gráfico. Gráficos de contorno são uma maneira eficaz de exibir dados em um espaço 3D, e são amplamente utilizados em campos científicos e de engenharia.