Модель хищник - жертва. Анимация погони

import numpy as np
import matplotlib.pyplot as plt
import math
 
# Начальные значения жертвы
x = 30
y = 15
v = 1
# Начальные значения хищника
X = 30
Y = 25
V = 5
# Параметры жертвы
v_ = 20
r_ = 5
b = 1
# Параметры хищника
V_ = 1
V__ = 15
R_ = 5
B = 1
step = 0.001  # шаг дельта
 
 
# Изменяемые по времени параметры
# Система уравнений
 
def R():
    return math.sqrt((x - X) ** 2 + (y - Y) ** 2)  # расстояние между хищником и жертвой (меняется постоянно)
 
 
def a():
    return b * (R() - r_)
 
 
def p():
    return ((x - X) + (y - Y)) / ((math.sqrt(2)) * R())
 
 
def q():
    return (- (x - X) + (y - Y)) / ((math.sqrt(2)) * R())
 
 
def A():
    return B * (R() - R_)
 
 
def P():
    return (x - X) / R()
 
 
def Q():
    return (y - Y) / R()
 
 
def dvdt():
    return a() * v * (v - v_)
 
 
def dxdt():
    return p() * v
 
 
def dydt():
    return q() * v
 
 
def dVdt():
    return A() * (V - V_) * (V - V__)
 
 
def dXdt():
    return P() * v
 
 
def dYdt():
    return Q() * v
 
 
mas_x = []
mas_y = []
mas_v = []
mas_X = []
mas_Y = []
mas_V = []
 
i = 0
while i < 1000:
    i = i + 1
    x = x + (dxdt() * step)
    y = y + (dydt() * step)
    v = v + (dvdt() * step)
    X = X + (dXdt() * step)
    Y = Y + (dYdt() * step)
    V = V + (dVdt() * step)
    mas_x.append(x)
    mas_y.append(y)
    mas_v.append(v)
    mas_X.append(X)
    mas_Y.append(Y)
    mas_V.append(V)
 
# следующие значения v, x, y жертвы и X, V, Y хищника вычисляются в цикле
# новое значение = предыдущее + (новое * шаг дельта)
 
fig = plt.figure()
# Добавление на рисунок прямоугольной (по умолчанию) области рисования
scatter2 = plt.scatter(mas_X, mas_Y)
scatter1 = plt.scatter(mas_x, mas_y)
plt.scatter(mas_X, mas_Y)
print('Scatter: ', type(scatter1))
 
plt.show()