Crear un gráfico de espiral con Python y Matplotlib

Beginner

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

Introducción

En este laboratorio, usaremos Python y Matplotlib para crear un gráfico de espiral relleno. Usaremos las bibliotecas numpy y matplotlib.pyplot para generar el gráfico.

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

Importaremos las bibliotecas necesarias para generar el gráfico. Usaremos numpy para cálculos numéricos y matplotlib.pyplot para crear el gráfico.

import matplotlib.pyplot as plt
import numpy as np

Definir variables

Definiremos las variables theta, a y b para nuestro gráfico.

theta = np.arange(0, 8*np.pi, 0.1)
a = 1
b =.2

Generar el gráfico

Generaremos el gráfico recorriendo cuatro valores de dt y concatenando las matrices resultantes de x e y.

for dt in np.arange(0, 2*np.pi, np.pi/2.0):

    x = a*np.cos(theta + dt)*np.exp(b*theta)
    y = a*np.sin(theta + dt)*np.exp(b*theta)

    dt = dt + np.pi/4.0

    x2 = a*np.cos(theta + dt)*np.exp(b*theta)
    y2 = a*np.sin(theta + dt)*np.exp(b*theta)

    xf = np.concatenate((x, x2[::-1]))
    yf = np.concatenate((y, y2[::-1]))

    p1 = plt.fill(xf, yf)

plt.show()

Interpretación

El gráfico generado por el código muestra una espiral que está llena de color. Las variables a y b controlan la forma de la espiral, mientras que la variable theta controla el número de revoluciones. El bucle a través de dt nos permite crear una espiral con cuatro "brazo" rotando el gráfico 45 grados cada vez.

Resumen

En este laboratorio, aprendimos cómo usar Python y Matplotlib para generar un gráfico de espiral relleno. Definimos variables para el gráfico, generamos el gráfico usando un bucle e interpretamos el gráfico resultante. Con este conocimiento, podemos crear gráficos similares con diferentes formas y colores.