Dibujo de contornos 3D con Matplotlib

Introducción

Este tutorial lo guiará a través del proceso de creación de un diagrama de contorno 3D utilizando Matplotlib en Python. El diagrama de contorno representa una superficie 3D utilizando contornos o curvas de nivel. Utilizaremos la función contour() para crear estas curvas de nivel y la opción extend3d=True para extender las curvas verticalmente en "cintas".

Consejos sobre la VM

Una vez finalizada la inicialización de 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 operaciones no puede automatizarse debido a las limitaciones de Jupyter Notebook.

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

Skills Graph

Importar bibliotecas

Debemos comenzar importando las bibliotecas necesarias para este tutorial. Utilizaremos matplotlib.pyplot para graficar, matplotlib.cm para mapas de colores y mpl_toolkits.mplot3d para graficación 3D.

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

Crear datos

A continuación, debemos crear los datos que utilizaremos para generar el diagrama de contorno. Utilizaremos la función get_test_data() del módulo mpl_toolkits.mplot3d para generar datos de muestra.

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

Crear ejes 3D

Utilizaremos la función add_subplot() para crear un subdiagrama 3D para nuestra gráfica. También estableceremos la proyección en '3d'.

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

Crear el diagrama de contorno

Ahora crearemos el diagrama de contorno utilizando la función contour(). Pasaremos los datos X, Y y Z, y estableceremos extend3d=True para extender las curvas verticalmente en "cintas". También estableceremos el mapa de colores en cm.coolwarm para un bonito esquema de colores.

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

Mostrar la gráfica

Finalmente, utilizaremos la función show() para mostrar nuestra gráfica.

plt.show()

Resumen

En este tutorial, aprendimos cómo crear un diagrama de contorno 3D utilizando Matplotlib en Python. Utilizamos la función contour() para crear curvas de nivel y la opción extend3d=True para extender las curvas verticalmente en "cintas". También utilizamos la función get_test_data() para generar datos de muestra y el mapa de colores cm.coolwarm para un bonito esquema de colores.