Diagrama de tallos tridimensional

Introducción

Esta práctica demuestra cómo crear un diagrama de tallos tridimensional utilizando la librería Matplotlib en Python. Un diagrama de tallos es una forma de representar puntos de datos dibujando líneas verticales desde una línea base hasta el punto de datos y colocando un marcador en la punta.

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 se puede automatizar 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 las bibliotecas necesarias

En este paso, importaremos las bibliotecas Matplotlib y Numpy utilizando la declaración import.

import matplotlib.pyplot as plt
import numpy as np

Definir los datos

En este paso, definiremos los datos que utilizaremos para crear el diagrama de tallos tridimensional. Crearemos una matriz linspace para el ángulo y utilizaremos las funciones seno y coseno para calcular las coordenadas x e y. También definiremos la coordenada z como el ángulo.

theta = np.linspace(0, 2*np.pi)
x = np.cos(theta - np.pi/2)
y = np.sin(theta - np.pi/2)
z = theta

Crear el diagrama de tallos tridimensional

En este paso, crearemos el diagrama de tallos tridimensional utilizando la función stem de Matplotlib. Pasaremos las coordenadas x, y y z como argumentos a la función stem.

fig, ax = plt.subplots(subplot_kw=dict(projection='3d'))
ax.stem(x, y, z)

plt.show()

Personalizar la gráfica

En este paso, personalizaremos el diagrama de tallos tridimensional cambiando la línea base utilizando el parámetro bottom y cambiando el formato utilizando los parámetros linefmt, markerfmt y basefmt.

fig, ax = plt.subplots(subplot_kw=dict(projection='3d'))
markerline, stemlines, baseline = ax.stem(
    x, y, z, linefmt='grey', markerfmt='D', bottom=np.pi)
markerline.set_markerfacecolor('none')

plt.show()

Cambiar la orientación de la gráfica

En este paso, cambiaremos la orientación de la gráfica utilizando el parámetro orientation. Estableceremos la orientación en 'x' para que los tallos se proyecten a lo largo de la dirección x y la línea base se encuentre en el plano yz.

fig, ax = plt.subplots(subplot_kw=dict(projection='3d'))
markerline, stemlines, baseline = ax.stem(x, y, z, bottom=-1, orientation='x')
ax.set(xlabel='x', ylabel='y', zlabel='z')

plt.show()

Resumen

En este laboratorio, aprendimos cómo crear un diagrama de tallos tridimensional utilizando la biblioteca Matplotlib en Python. Comenzamos definiendo los datos y luego creamos la gráfica utilizando la función stem. También personalizamos la gráfica cambiando el formato y la orientación.