Creación de visualizaciones de contornos 3D

Introducción

En este laboratorio, aprenderemos a crear un gráfico de contorno 3D utilizando la biblioteca Matplotlib en Python. Un gráfico de contorno es una representación gráfica de una superficie 3D en la que se trazan contornos en un plano 2D. Los gráficos de contorno son útiles para visualizar la variación de una variable con respecto a otras dos variables.

Consejos sobre la VM

Una vez que se haya iniciado la VM, haga clic en la esquina superior izquierda para cambiar a la pestaña Cuaderno y acceder a Jupyter Notebook para practicar.

A veces, es posible que tenga que esperar unos segundos a que Jupyter Notebook termine de cargarse. La validación de las operaciones no se puede automatizar debido a las limitaciones de Jupyter Notebook.

Si tiene problemas durante el aprendizaje, no dude en preguntar a Labby. Deje sus comentarios después de la sesión y lo resolveremos rápidamente para usted.

Importar bibliotecas

Comenzaremos importando las bibliotecas necesarias para crear el gráfico de contorno 3D. Utilizaremos las bibliotecas matplotlib y mpl_toolkits.

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

Crear figura y subtrama

A continuación, creamos una figura y una subtrama para contener nuestro gráfico de contorno 3D.

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

Obtener datos de prueba

Utilizaremos la función axes3d.get_test_data() para obtener algunos datos de prueba para graficar.

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

Crear gráfico de contorno

Ahora podemos crear un gráfico de contorno 3D de los datos de prueba utilizando la función ax.contour().

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

Personalizar el gráfico

Podemos personalizar el gráfico agregando etiquetas a los ejes y ajustando el ángulo de visualización.

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

Mostrar el gráfico

Finalmente, utilizamos la función plt.show() para mostrar el gráfico de contorno 3D.

plt.show()

Resumen

En este laboratorio, aprendimos cómo crear un gráfico de contorno 3D utilizando la librería Matplotlib en Python. Importamos las bibliotecas necesarias, creamos una figura y un subgráfico, obtuvimos datos de prueba, creamos un gráfico de contorno, personalizamos el gráfico y mostramos el gráfico. Los gráficos de contorno son útiles para visualizar la variación de una variable con respecto a otras dos variables.