Criar Gráfico de Espiral com Python e Matplotlib

Beginner

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

Introdução

Neste laboratório, usaremos Python e Matplotlib para criar um gráfico de espiral preenchida. Usaremos as bibliotecas numpy e matplotlib.pyplot para gerar o gráfico.

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

Importaremos as bibliotecas necessárias para gerar o gráfico. Usaremos numpy para cálculos numéricos e matplotlib.pyplot para criar o gráfico.

import matplotlib.pyplot as plt
import numpy as np

Definir variáveis

Definiremos as variáveis theta, a e b para nosso gráfico.

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

Gerar o gráfico

Geraremos o gráfico iterando por quatro valores de dt e concatenando os arrays x e y resultantes.

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()

Interpretação

O gráfico gerado pelo código mostra uma espiral preenchida com cor. As variáveis a e b controlam a forma da espiral, enquanto a variável theta controla o número de revoluções. O loop através de dt nos permite criar uma espiral com quatro "braços" rotacionando o gráfico em 45 graus a cada vez.

Resumo

Neste laboratório, aprendemos como usar Python e Matplotlib para gerar um gráfico de espiral preenchida. Definimos variáveis para o gráfico, geramos o gráfico usando um loop e interpretamos o gráfico resultante. Com este conhecimento, podemos criar gráficos semelhantes com diferentes formas e cores.