Как построить только текущий кадр с анимацией. FuncAnimation?

Question

Генерация данных:

import glob
import pandas as pd
import numpy as np

coords = dict()
for j in range(10):
    x=np.random.uniform(0.0, 5.0,2);
    y=np.random.uniform(0.0, 5.0,2);
    vx=np.random.uniform(-1,1,2);
    vy=np.random.uniform(-1,1,2);
    coords[j]= {'x': x, 'y':y, 'vx':vx, 'vy':vy}

Участок черчения:

import numpy as np
from matplotlib import pyplot as plt
from matplotlib import animation
from matplotlib import animation, rc
from IPython.display import HTML
%matplotlib inline

fig, ax = plt.subplots(1,1)


ax.set_xlim(-5, 5)
ax.set_ylim(-5, 5)

def update_quiver(frameIdx):
    global coords
    frame = coords[frameIdx]
    X = frame['x']
    Y = frame['y']
    U = frame['vx']
    V = frame['vy']
    Q = ax.quiver(X, Y, U, V, pivot='mid', color='k', units='inches')

    Q.set_UVC(U,V)
    return Q,

А затем черчение:

rc('animation', html='jshtml')
anim = animation.FuncAnimation(fig, update_quiver, fargs=(),
                               interval=50, blit=False, frames=max(coords.keys()))
anim

Как я могу визуализировать только текущее местоположение частиц, а не всю траекторию?

Когда я пытаюсь:

def update_quiver(frameIdx):
    global coords
    fig.clf()
    ax.set_xlim(-5, 5)
    ax.set_ylim(-5, 5)
    frame = coords[frameIdx]
    X = frame['x']
    Y = frame['y']
    U = frame['vx']
    V = frame['vy']
    Q = ax.quiver(X, Y, U, V, pivot='mid', color='k', units='inches')


    Q.set_UVC(U,V)
    return Q,

Это ничего не показывает.

Answer 1

Возможность удалить колчан, о котором я прокомментировал, может выглядеть следующим образом:

import numpy as np; np.random.seed(1)

coords = dict()
for j in range(10):
    x=np.random.uniform(0.0, 5.0,2);
    y=np.random.uniform(0.0, 5.0,2);
    vx=np.random.uniform(-1,1,2);
    vy=np.random.uniform(-1,1,2);
    coords[j]= {'x': x, 'y':y, 'vx':vx, 'vy':vy}

import numpy as np
from matplotlib import pyplot as plt, animation

fig, ax = plt.subplots(1,1)

ax.set_xlim(-1, 6)
ax.set_ylim(-1, 6)

Q = ax.quiver(1,1,1,1, alpha=0)

def update_quiver(frameIdx):
    global coords, Q
    frame = coords[frameIdx]
    X = frame['x']
    Y = frame['y']
    U = frame['vx']
    V = frame['vy']
    Q.remove()
    Q = ax.quiver(X, Y, U, V, pivot='mid', color='k', units='inches')
    return Q,


anim = animation.FuncAnimation(fig, update_quiver, fargs=(),
                               interval=500, blit=False, frames=max(coords.keys()))

plt.show()

Answer 2

Вы можете попробовать очистить цифру в update_quiver:

fig.clf()

РЕДАКТИРОВАТЬ: этот код работает на моем компьютере:

def update_quiver(frameIdx):
    frame = coords[frameIdx]

    plt.cla()
    ax.set_xlim(-1, 7)
    ax.set_ylim(-1, 7)

    X = frame['x']
    Y = frame['y']
    U = frame['vx']
    V = frame['vy']
    Q = ax.quiver(X, Y, U, V, pivot='mid', color='k', units='inches')

    Q.set_UVC(U,V)
    return Q,

anim = animation.FuncAnimation(fig, update_quiver, fargs=(),interval=500, blit=True,frames=np.arange(len(coords)))
fig.tight_layout()
plt.show()