Harmonógrafo-Animado3D.py

import matplotlib.animation as animation
from matplotlib.animation import PillowWriter
from IPython.display import HTML, Image
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np
plt.style.use(['dark_background'])

#Definindo a função pendulo com valor de parâmetros padrão
def Pendulo(t, A = 1, f = 5, p = 0, d = 0.05):
    return (A*np.sin(2*np.pi*f*t + p)*np.exp(-1*d*t))

#Criando um array para t
t = np.linspace(0, 10*np.pi, 4500)

#Plotando os resultados
fig = plt.figure(figsize = (12, 6), dpi = 100)

ax = fig.add_subplot(projection = '3d')
ax.view_init(30, -10)#20,-20

ax.w_xaxis.pane.fill = False
ax.w_yaxis.pane.fill = False
ax.w_zaxis.pane.fill = False

xt = np.zeros(len(t))
yt = np.zeros(len(t))
zt = np.zeros(len(t))

#Plotando a animação
def update(i):
    xt[i] = Pendulo(t[i], 2, 4, -1, 0.05) + Pendulo(t[i], 1, 2, np.pi, 0.01)
    yt[i] = Pendulo(t[i], 4, 2, np.pi/4, 0.02) + Pendulo(t[i], 1, 2, np.pi/2, 0.25)
    zt[i] = Pendulo(t[i])
    ax.view_init(30, -10+i/2)
    ax.clear()
    ax.plot(xt[:i], yt[:i], zt[:i], 'green')
    ax.set_facecolor('k')
    # ax.grid(False)
    ax.set_title('$Harmonógrafo$: t = ' + str(i), color = 'g')
    ax.set_xlabel('$x(t)$', color = 'g')
    ax.set_ylabel('$y(t)$', color = 'g')
    ax.set_zlabel('$z(t)$', color = 'g')

#Plotagem para chamar a função update() periodicamente
ani = animation.FuncAnimation(fig, update, np.arange(1000), interval = 100, repeat = False)
#ani.save('Harmonógrafo3D.gif', writer = PillowWriter(fps = 60))
#ani.save('Harmonógrafo3D.mp4', writer = 'ffmpeg', fps = 60)
plt.show()