[DOC]: Add simple animation scatter plot to the example documentation (#24096)

kostyafarber · web-flow · commit 1ade85c7d176 · 2022-10-20T17:55:28.000-04:00
* add simple animation scatter plot

* change title to Animated Scatter Saved as Gif

* change title to sentence case
diff --git a/examples/animation/simple_scatter.py b/examples/animation/simple_scatter.py
@@ -0,0 +1,31 @@
+"""
+=============================
+Animated scatter saved as GIF
+=============================
+
+"""
+import numpy as np
+import matplotlib.pyplot as plt
+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()
