python matplotlib 播放图片变化过程

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import animation
from PIL import Image

img_ori = np.array(Image.open(‘5.png‘))
img_tran = np.load(‘filename.npy‘)
fig = plt.figure()
plt.subplot(1, 2, 1)
plt.imshow(img_ori)
plt.subplot(1, 2, 2)
ims = []
i = [i for i in range(200) if i % 5 == 0]
def update(i):
    return plt.imshow(img_tran[i].reshape([28, 28]))
ani = animation.FuncAnimation(fig, update)
ani.save(‘contrast.mp4‘, fps=15,
                 extra_args=[‘-vcodec‘, ‘libx264‘],
                 writer=‘ffmpeg_file‘)
plt.show()