Задача с FuncAnimation для создания анимации маятникового маятника - PullRequest
0 голосов
/ 08 ноября 2019

Проблема, которую я моделирую, - это простой маятник. Хотя я делал это до использования PyGame, теперь я решил использовать инструменты анимации matplotlib. Это работает, но не с желаемым эффектом. Кажется, что имитация в реальном времени работает. Я настроил интервал и количество кадров, но частота кадров слишком низкая. Как вы увеличиваете частоту кадров, пока играете в реальном времени. Буду весьма признателен за это. В любом случае вот мой код:

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

g = 9.80665
L = 2
mu = 0.1
t = 100
theta_0 = np.pi/3
d_theta_0 = 0

def get_d2_theta(theta,d_theta):
    return -mu*d_theta-(g/L)*np.sin(theta)

def theta(t):
    theta = theta_0
    d_theta = d_theta_0
    delta_t = 1./60
    for time in np.arange(0,t,delta_t): 
        d2_theta = get_d2_theta(theta,d_theta)
        theta += d_theta*delta_t
        d_theta += d2_theta*delta_t
    return theta

x_data = [0,0]
y_data = [0,0]

fig, ax = plt.subplots()
ax.set_xlim(-2, 2)
ax.set_ylim(-2.5,1)
line, = ax.plot(0, 0)

def animation_frame(i):
    x = L*np.sin(theta(i))
    y = -L*np.cos(theta(i))

    x_data[1] = x
    y_data[1] = y

    line.set_xdata(x_data)
    line.set_ydata(y_data)
    return line, 

animation = FuncAnimation(fig, func=animation_frame, frames=np.arange(0, 60, (1./60)),interval = 10)
plt.show()

...