"""

=============================

Animated scatter saved as GIF

=============================

Output generated via `matplotlib.animation.Animation.to_jshtml`.

"""

import matplotlib.pyplot as plt

import numpy as np

import matplotlib.animation as animation

fig, ax = plt.subplots()

ax.set_xlim(0, 10)

scat = ax.scatter(1, 0)

x = np.linspace(0, 10)

def animate(i):

scat.set_offsets((x[i], 0))

return (scat,)

ani = animation.FuncAnimation(fig, animate, repeat=True, frames=len(x) - 1, interval=50)

# To save the animation using Pillow as a gif

# writer = animation.PillowWriter(fps=15,

# metadata=dict(artist='Me'),

# bitrate=1800)

# ani.save('scatter.gif', writer=writer)

plt.show()

# %%

#

# .. tags::

# component: animation,

# plot-type: scatter,

# purpose: reference,

# level: intermediate