Как анимировать цикл for с переменным диапазоном

Question

Я получаю разные графики для каждого значения, которое я записал для верхнего диапазона цикла, как я и ожидал. Но я бы хотел анимировать график из диапазона (0,0) и перейти к (0,15) при изменении верхнего предела 1 на 1, используя функцию анимации matplotlib animation.FuncAnimation(). Таким образом, в анимации будет всего 16 кадров. Я испортил часть анимации, поэтому я вставляю код, который дает 1 вывод сюжета. Заранее спасибо!

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

g = 1.0

def distance(x1,y1,x2,y2):
    r = np.sqrt((x2-x1)**2+(y2-y1)**2)
    return r

def gravit(m1,m2,r):
    f = g*m1*m2/(r**2)
    return f

def angle(y2, y1, x2, x1):
    ydif = y1-y2
    xdif = x1-x2
    angle = np.arctan2(ydif,xdif)
    return angle

m1 = 100
x1, y1 = 0,0
m2 = 1
x2, y2 = -15,-10
vx1 = 1
vy1 = 0
ax1 = 0
ay1 = 0
vx2 = 2
vy2 = 3.9
ax2 = 0
ay2 = 0
x1coor = [x1]
y1coor = [y1]
x2coor = [x2]
y2coor = [y2]


for t in range(0,10):    #This value of 10 should be change from 0 to 15 in the animation
    r = distance(x1,y1,x2,y2)    
    fx1 = gravit(m1, m2, r) * np.cos(angle(y2,y1,x2,x1))
    fy1 = gravit(m1, m2, r) * np.sin(angle(y2,y1,x2,x1))
    ax2 = fx1/m2
    vx2 = vx2 + ax2
    ay2 = fy1/m2
    vy2 = vy2 + ay2
    x2 = x2 + vx2 + 0.5*ax2
    y2 = y2 + vy2 + 0.5*ay2
    x1 = x1 + vx1 + 0.5*ax1
    y1 = y1 + vy1 + 0.5*ay1
    x1coor.append(x1)
    y1coor.append(y1)
    x2coor.append(x2)
    y2coor.append(y2)

plt.axes().set_aspect('equal')
plt.axis([-30,30,-30,30])
plt.plot(x1coor,y1coor, '-.', color='blue')
plt.plot(x2coor,y2coor, '-.', color='black')
plt.scatter(x1,y1,s=m1*20, color='blue')
plt.scatter(x2,y2,s=m2*20, color='red')

Answer 1

Функция FuncAnimation модуля анимации matplotlib требует фигуры и функции для рисования каждого кадра, поэтому сначала инициализируйте фигуру следующим образом:

fig = plt.figure()
ax1 = fig.add_subplot(1, 1, 1)

ax1.set_aspect("equal")
ax1.set_xlim(-30, 30)
ax1.set_ylim(-30, 30)
l1, = ax1.plot([], [], linestyle = "-.", color = "blue")
l2, = ax1.plot([], [], '-.', color = "black")
s1, = ax1.plot([], [], linestyle = None, marker = "o", markersize = 5, color = "blue")
s2, = ax1.plot([], [], linestyle = None, marker = "o", markersize = 5, color = "red")

Обратите внимание, что различные линейные графики не содержат входных данных и сохраняются в переменных l1, l2, s1 и s2.

Теперь определите функцию, которая будет вызываться для рисования каждого кадра. Первым аргументом всегда является номер кадра. Эта функция должна возвращать все объекты графика, которые были обновлены с использованием их методов set_data:

def update_fig(i, x1coor, y1coor, x2coor, y2coor):
    l1.set_data(x1coor[:i+1], y1coor[:i+1])
    l2.set_data(x2coor[:i+1], y2coor[:i+1])
    s1.set_data(x1coor[i], y1coor[i])
    s2.set_data(x2coor[i], y2coor[i])
    return l1, l2, s1, s2,

Теперь вы можете запустить анимацию (обратите внимание, что аргументы update_fig, кроме «i», передаются в функцию с помощью ключевого слова «fargs»):

ani = animation.FuncAnimation(fig, update_fig,
                              frames = len(x1coor),
                              fargs = (x1coor, y1coor, x2coor, y2coor),
                              interval = 100,
                              repeat = True,
                              )