python使用matplotlib动画

使用matplotlib画动画。

import matplotlib
matplotlib.use('TkAgg')  #必须有， 不然不会实时显示
import matplotlib.pyplot as plt
from skimage import io
import os

image_dir= './images'

image_num = len(os.listdir(image_dir))

plt.ion()  # 开启交互式模式， 必须有， 不然不会实时显示
fig = plt.figure()
ax1 = fig.add_subplot(111, aspect='equal')  #添加子图。 推荐使用子图， 直接使用plt.imshow()会出现图关不掉的情况

for index in range(image_num):
    image_name = "{:04}.jpg".format(index)
    image_file = os.path.join(image_dir, image_name)
    image = io.imread(image_file)

    ax1.imshow(image)

    fig.canvas.flush_events() # 保存 GUI 事件,直到 UI 事件被处理。 必须有，不然图像一闪而过
    plt.draw()
    ax1.cla()  #清除坐标区并重置所有坐标区属性