Анимация квадрата (Matplotlib)

import matplotlib.pyplot as plt
from matplotlib.patches import Rectangle
import matplotlib.patches as patches
 
 
plt.xlim(0, 13)
plt.ylim(0, 13)
 
ax = plt.gca()
patch = patches.Rectangle((0, 0), 0.5, 0.5, fc='green')
ax.add_patch(patch)
plt.grid(True)
plt.show()

import matplotlib.pyplot as plt
import matplotlib.patches as patches
from matplotlib import animation
 
X = [ 1, 2, 3, 4, 5, 6   ]
Y = [ 2, 3, 5, 7, 11, 13 ]
w = 0.5   # сторона квадрата 
fig = plt.figure()
plt.grid()
ax = plt.axes( xlim = ( min(X), max(X) + w ), 
               ylim = ( min(Y), max(Y) + w )  )
ax.set_aspect( 'equal' )
patch = patches.Rectangle( (0, 0), w, w, 0 )
 
def init():
    ax.add_patch( patch )
    return patch,
 
def animate( i ):
    patch.set_xy( [X[i], Y[i] ] )
    return patch,
 
anim = animation.FuncAnimation( fig, animate,
                                init_func = init,
                                frames    = len( X ),
                                interval  = 1000,
                                blit      = True     )
plt.show()