add simple animation scatter plot

kostyafarber · kostyafarber · commit dde1897fb04a · 2022-10-05T13:06:51.000+01:00
diff --git a/examples/animation/simple_scatter.py b/examples/animation/simple_scatter.py
@@ -0,0 +1,31 @@
+"""
+======================
+Animated scatter plot
+======================
+
+"""
+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 gifs
+# writer = animation.PillowWriter(fps=15,
+#                                 metadata=dict(artist='Me'),
+#                                 bitrate=1800)
+# ani.save('scatter.gif', writer=writer)
+
+plt.show()
