DOC: change artists, add annotation

jklymak · jklymak · commit 449a57c2b564 · 2023-01-09T17:03:23.000-08:00
diff --git a/tutorials/introductory/animation_tutorial.py b/tutorials/introductory/animation_tutorial.py
@@ -89,30 +89,36 @@
 # tutorial but can be found in their respective documentations. An example of
 # such update methods in use for `.Axes.scatter` is as follows.
 
-
 fig, ax = plt.subplots()
-t = np.linspace(-np.pi, np.pi, 400)
-a, b = 3, 2
-delta = np.pi / 2
+t = np.linspace(0, 3, 40)
+g = -9.81
+v0 = 12
+z = g * t**2 / 2 + v0 * t
+
+v02 = 5
+z2 = g * t**2 / 2 + v02 * t
 
-scat = ax.scatter(np.sin(a * t[0] + delta), np.sin(b * t[0]), c="b", s=2)
-ax.set_xlim(-1.5, 1.5)
-ax.set_ylim(-1.5, 1.5)
+scat = ax.scatter(t[0], z[0], c="b", s=5, label=f'v0 = {v0} m/s')
+line2 = ax.plot(t[0], z2[0], label=f'v0 = {v02} m/s')[0]
+ax.set(xlim= [0, 3], ylim=[-4, 10], xlabel='Time [s]', ylabel='Z [m]')
+ax.legend()
 
 
 def update(frame):
     # .set_offsets replaces the offset data for the entire collection with
     # the new values. Therefore, to also carry forward the previously
     # calculated information, we use the data from the first to the current
     # frame to set the new offsets.
-    x = np.sin(a * t[:frame] + delta)
-    y = np.sin(b * t[:frame])
+    x = t[:frame]
+    y = z[:frame]
     data = np.stack([x, y]).T
     scat.set_offsets(data)
-    return (scat,)
+    line2.set_xdata(t[:frame])
+    line2.set_ydata(z2[:frame])
+    return (scat, line2)
 
 
-ani = animation.FuncAnimation(fig=fig, func=update, frames=40, interval=300)
+ani = animation.FuncAnimation(fig=fig, func=update, frames=40, interval=30)
 plt.show()
 
 
