Me gustaría tener un gráfico trazado de forma iterativa que permita saltar al siguiente cuadro, detenerlo y volver al cuadro anterior.

He mirado el módulo de animación matplotlib que sería perfecto si hubiera una manera de implementar la funcionalidad del cuadro anterior (como ejecutar la animación hacia atrás durante unos cuadros cuando se presiona una tecla)

Sería bueno hacer algo como esto:

def update_frame(i, data):
    fig.set_data(data[i])

pero de una manera que podría gestionar explícitamente si el iterador aumenta o disminuye.

¿Hay alguna manera de hacer eso en matplotlib? ¿Debo buscar un módulo de Python diferente?