julia_matplotlib_ex.py

# same as the Julia example, but in Python
# I hope I made no mistake (mistaking an attribute as a
# function or something like this while converting it
# from the Julia to Python. Didn't test it)

def draw_new_frame(fig, cb, image, frame, file)
    image.set_data(frame)

    # get the filename written prev.
    texts = fig.texts
    for i in range(len(texts)):
        # and remove each
        texts[1].remove()
    # write new filename
    fig.text(0.4, 0.9, file)
    # scale the colorbar
    #image[:autoscale]()
    image.set_clim(0, 100)
    cb.update_normal(image)
    fig.canvas.draw()
    plt.draw()