Анимация квадратичных изменений сетки (matshow)

У меня есть сетка NxN с некоторыми значениями, которые меняются каждый временной шаг. . Я нашел способ построить единую конфигурацию сетки с помощью функции matshow, но я не знаю, как мне обновлять статус с каждым временным шагом. Вот простой пример:

from pylab import *
from matplotlib import pyplot

a = arange(25)
a = a.reshape(5,5)
b = 10*rand(5,5)
matshow(a-b, cmap = cm.jet)
colorbar()
show()

Этот код выдает следующую картинку:
Теперь представьте, что на следующем временном шаге некоторые значения меняются, то же самое должно происходить и на этой картинке. Вот логика, которую я имел в виду:

from pylab import *
from matplotlib import pyplot

a = arange(25)
a = a.reshape(5,5)
time=10
for t in range(time):
    b = 10*rand(5,5)
    print b
    matshow(a-b, cmap=cm.jet)
    colorbar()
show()

Получается 10 картинок. Я хотел бы анимировать это вместо того, чтобы создавать отдельные изображения, и, например, я хотел бы выбрать временной шаг между изменениями (то есть частоту кадров).
Кроме того, я открыт для предложений по другим функциям, если matshowне подходит, но, пожалуйста, будьте проще, я относительно неопытен.