im = ax.imshow(mat, interpolation='nearest')
with writer.saving(fig, 'writer_test.mp4', 100):
    for i in range(100):
        mat = np.random.random((100, 100))
im.set_data(mat)
writer.grab_frame()
Matplotlib.animation: как удалить белое поле