Simplify examples by removing data gen functions

chahak13 · chahak13 · commit 88e3bcc86ee3 · 2022-10-27T12:22:36.000-05:00
diff --git a/tutorials/introductory/animation_tutorial.py b/tutorials/introductory/animation_tutorial.py
@@ -20,13 +20,6 @@
 # :class:`~matplotlib.animation.FuncAnimation` uses a user-provided function by
 # repeatedly calling the function with at a regular *interval*. This allows
 # dynamic generation of data by using generators.
-import itertools
-
-
-def data_gen():
-    for cnt in itertools.count():
-        t = cnt/10
-        yield t, np.sin(2 * np.pi * t)
 
 
 def init():
@@ -43,21 +36,20 @@ def init():
 xdata, ydata = [], []
 
 
-def update(data):
-    t, y = data
-    xdata.append(t)
-    ydata.append(y)
+def update(frame):
+    xdata.append(frame)
+    ydata.append(np.sin(2 * np.pi * frame / 30))
     xmin, xmax = ax.get_xlim()
 
-    if t >= xmax:
+    if frame >= xmax:
         ax.set_xlim(xmin, 2*xmax)
         ax.figure.canvas.draw()
     line.set_data(xdata, ydata)
 
     return line,
 
 ani = animation.FuncAnimation(
-    fig=fig, func=update, frames=data_gen, init_func=init, interval=20
+    fig=fig, func=update, frames=240, init_func=init, interval=30
 )
 plt.show()
 
@@ -73,25 +65,18 @@ def update(data):
     rng.uniform(low=0, high=1, size=100),
     c='b'
 )
+ax.grid()
 
 
-def data_gen():
-    for cnt in itertools.count():
-        x, y = (
-            rng.uniform(low=0, high=1, size=100),
-            rng.uniform(low=0, high=1, size=100)
-            )
-        yield cnt, x, y
-
-
-def update(inputs):
-    frame, x, y = inputs
+def update(frame):
+    x = rng.uniform(low=0, high=1, size=100)
+    y = rng.uniform(low=0, high=1, size=100)
     data = np.stack([x, y]).T
     scat.set_offsets(data)
     return scat,
 
 ani = animation.FuncAnimation(
-    fig=fig, func=update, frames=data_gen, interval=200
+    fig=fig, func=update, frames=240, interval=300
 )
 plt.show()
 
@@ -103,22 +88,16 @@ def update(inputs):
 fig, ax = plt.subplots()
 rng = np.random.default_rng()
 
-aximg = ax.imshow(rng.uniform(low=0, high=1, size=(100, 100)))
-
+aximg = ax.imshow(rng.uniform(low=0, high=1, size=(10, 10)), cmap="Blues")
 
-def data_gen():
-    for cnt in itertools.count():
-        x = rng.uniform(low=0, high=1, size=(100, 100))
-        yield cnt, x
 
-
-def update(inputs):
-    frame, x = inputs
-    aximg.set_data(x)
+def update(frame):
+    data = rng.uniform(low=0, high=1, size=(10, 10))
+    aximg.set_data(data)
     return aximg,
 
 ani = animation.FuncAnimation(
-    fig=fig, func=update, frames=data_gen, interval=200
+    fig=fig, func=update, frames=240, interval=200
 )
 plt.show()
 
@@ -131,13 +110,14 @@ def update(inputs):
 
 
 fig, ax = plt.subplots()
+ax.grid()
 rng = np.random.default_rng()
 
 x_frames = rng.uniform(low=0, high=1, size=(100, 120))
 y_frames = rng.uniform(low=0, high=1, size=(100, 120))
 artists = [
     [
-        ax.scatter(x_frames[:, i], y_frames[:, i])
+        ax.scatter(x_frames[:, i], y_frames[:, i], c="b")
     ]
     for i in range(x_frames.shape[-1])
 ]
@@ -150,6 +130,7 @@ def update(inputs):
 # -------------------------------------------
 
 fig, ax = plt.subplots()
+ax.grid()
 rng = np.random.default_rng()
 
 scat = ax.scatter(
@@ -159,23 +140,15 @@ def update(inputs):
 )
 
 
-def data_gen():
-    for cnt in itertools.count():
-        x, y = (
-            rng.uniform(low=0, high=1, size=100),
-            rng.uniform(low=0, high=1, size=100)
-            )
-        yield cnt, x, y
-
-
-def update(inputs):
-    frame, x, y = inputs
+def update(frame):
+    x = rng.uniform(low=0, high=1, size=100)
+    y = rng.uniform(low=0, high=1, size=100)
     data = np.stack([x, y]).T
     scat.set_offsets(data)
     return scat,
 
 ani = animation.FuncAnimation(
-    fig=fig, func=update, frames=data_gen, interval=200
+    fig=fig, func=update, frames=240, interval=200
 )
 # ani.save(filename="/tmp/pillow_example.gif", writer="pillow")
 # ani.save(filename="/tmp/pillow_example.apng", writer="pillow")
