Как использовать анимацию Matplotlib для визуализации пользовательской среды Open AI Gym - PullRequest
Новая
       25

Как использовать анимацию Matplotlib для визуализации пользовательской среды Open AI Gym

0 голосов
/ 18 июня 2019

Я работаю над созданием новой среды OpenAI Gym Environment и хочу отобразить живую анимацию, которую я кодировал для отображения с помощью Matplotlib.

Я попытался построить и вызвать plt.show () вметод рендеринга, но при этом создается новое окно для каждого кадра, а старое окно закрывается для открытия нового.

Мой текущий код для создания анимации: 

fig = plt.figure(figsize=(10,10), dpi=80)

plt.axis([0, SIZE, 0, SIZE])
plt.xticks([])
plt.yticks([])

plt.hist2d(x,y, bins=10, range=[[0, SIZE], [0, SIZE]])
sc, = plt.plot(x, y, 'ro')

# Plot boids
def plot_boids(boids):
  #... compute some stuff here ...

  plt.hist2d(x,y, bins=10, range=[[0, SIZE], [0, SIZE]])
  return sc.set_data(x, y)

def animate(i):
  # ... compute some stuff here ...

  fig = plot_boids(boids)

anim=animation.FuncAnimation(fig, animate, repeat=False, 
                              blit=False, frames=100, interval=100)
plt.show()

Можно ли как-нибудь интегрировать метод анимации plt непосредственно в метод рендеринга для тренажерного зала?

...