Plotagem de Contorno 3D com Matplotlib

Beginner

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

Introdução

Este tutorial irá guiá-lo através do processo de criação de um gráfico de contorno 3D usando Matplotlib em Python. O gráfico de contorno representa uma superfície 3D usando contornos ou curvas de nível. Usaremos a função contour() para criar essas curvas de nível e a opção extend3d=True para estender as curvas verticalmente em 'fitas' (ribbons).

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.

Importar Bibliotecas

Precisamos começar importando as bibliotecas necessárias para este tutorial. Usaremos matplotlib.pyplot para plotagem, matplotlib.cm para mapas de cores e mpl_toolkits.mplot3d para plotagem 3D.

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

Criar Dados

Em seguida, precisamos criar os dados que usaremos para gerar o gráfico de contorno. Usaremos a função get_test_data() do módulo mpl_toolkits.mplot3d para gerar dados de exemplo.

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

Criar Eixos 3D

Usaremos a função add_subplot() para criar um subplot 3D para nosso gráfico. Também definiremos a projeção para '3d'.

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

Criar Gráfico de Contorno

Agora criaremos o gráfico de contorno usando a função contour(). Passaremos os dados X, Y e Z e definiremos extend3d=True para estender as curvas verticalmente em "fitas" (ribbons). Também definiremos o mapa de cores para cm.coolwarm para um esquema de cores agradável.

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

Exibir o Gráfico

Finalmente, usaremos a função show() para exibir nosso gráfico.

plt.show()

Resumo

Neste tutorial, aprendemos como criar um gráfico de contorno 3D usando Matplotlib em Python. Usamos a função contour() para criar curvas de nível e a opção extend3d=True para estender as curvas verticalmente em "fitas" (ribbons). Também usamos a função get_test_data() para gerar dados de exemplo e o mapa de cores cm.coolwarm para um esquema de cores agradável.