Introdução
Na análise de dados, é frequentemente necessário criar gráficos 3D para visualizar dados. No Matplotlib, podemos criar gráficos 3D como subplots para comparar diferentes dados 3D. Este laboratório demonstrará como criar gráficos 3D como subplots usando Matplotlib.
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
Antes de começarmos, precisamos importar as bibliotecas que usaremos neste laboratório. Usaremos Matplotlib, NumPy e Axes3D de mpl_toolkits.mplot3d.
import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d import Axes3D
Criar a Figura e os Subplots
Criaremos uma figura com dois subplots. O primeiro subplot será um gráfico de superfície 3D, e o segundo subplot será um gráfico wireframe 3D.
## Criar uma figura com dois subplots
fig = plt.figure(figsize=plt.figaspect(0.5))
## Adicionar o primeiro subplot com projeção 3D
ax1 = fig.add_subplot(1, 2, 1, projection='3d')
## Adicionar o segundo subplot com projeção 3D
ax2 = fig.add_subplot(1, 2, 2, projection='3d')
Criar o Gráfico de Superfície 3D
Criaremos um gráfico de superfície 3D para o primeiro subplot. Usaremos NumPy para criar os dados para o gráfico.
## Criar dados para o gráfico de superfície 3D
X = np.arange(-5, 5, 0.25)
Y = np.arange(-5, 5, 0.25)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
## Plotar o gráfico de superfície 3D
surf = ax1.plot_surface(X, Y, Z, cmap='coolwarm', linewidth=0, antialiased=False)
## Adicionar uma barra de cores ao gráfico
fig.colorbar(surf, shrink=0.5, aspect=10)
## Definir os limites para o eixo z
ax1.set_zlim(-1.01, 1.01)
Criar o Gráfico Wireframe 3D
Criaremos um gráfico wireframe 3D para o segundo subplot. Usaremos a função get_test_data de mpl_toolkits.mplot3d.axes3d para criar os dados para o gráfico.
## Criar dados para o gráfico wireframe 3D
X, Y, Z = Axes3D.get_test_data(0.05)
## Plotar o gráfico wireframe 3D
ax2.plot_wireframe(X, Y, Z, rstride=10, cstride=10)
Mostrar o Gráfico
Usaremos a função plt.show() para exibir o gráfico.
plt.show()
Resumo
Neste laboratório, aprendemos como criar gráficos 3D como subplots usando Matplotlib. Criamos uma figura com dois subplots, um gráfico de superfície 3D e um gráfico wireframe 3D. Usamos NumPy para criar os dados para o gráfico de superfície 3D e a função get_test_data de mpl_toolkits.mplot3d.axes3d para criar os dados para o gráfico wireframe 3D. Também adicionamos uma barra de cores ao gráfico de superfície 3D e definimos os limites para o eixo z.