Как я могу динамически (в цикле) показывать изображения в Google Colab?

Question

Я пытался использовать pyplot / matplotlib, чтобы показывать изображения по мере их изменения в цикле, но я не смог заставить что-либо работать.Я в основном не могу обновить показанное изображение.Вот код для решения проблемы:

f = plt.figure(1)
ax = plt.gca()
show_obj= ax.imshow(np.random.randn(28,28))
for i in range(10):
  print(i)
  # None of these 3 options work
  if True:
    # the image does not update
    show_obj.set_data(np.random.randn(28,28))
    f.canvas.draw()
  if False:
    # image does not update
    ax.clear()
    ax.imshow(np.random.rand(28,28))
    pass
  if False:
    # starts drawing new axes
    f = plt.figure(1)
    ax = plt.gca()
    ax.imshow(np.random.rand(28,28))
  plt.show()

Answer 1

Я тестирую этот код, и он работает.

from IPython.display import clear_output
import matplotlib.pyplot as plt
from numpy.random import randn
from time import sleep

for i in range(5):
  clear_output()
  plt.imshow(randn(28, 28))
  plt.show()
  sleep(1)