Update animated_FuncAnimation.py

vamshipy · web-flow · commit deac3e6770fc · 2024-01-20T10:11:52.000+05:30
diff --git a/galleries/examples/animation/animated_FuncAnimation.py b/galleries/examples/animation/animated_FuncAnimation.py
@@ -10,24 +10,24 @@
 
 # Constants
 decades = np.arange(1940, 2020, 10)
-initial_engineers = 10000  # Rough estimate of the number of software engineers(not real) in the 1950s
+initial_widgets = 10000  # Rough estimate of the no. of widgets in the 1950s
 
 # Generate rough growth data
 growth_rate = np.random.uniform(1.02, 3.10, size=len(decades))
-engineers_data = np.cumprod(growth_rate) * initial_engineers
+widgets_data = np.cumprod(growth_rate) * initial_widgets
 
 # Set up the initial plot
 fig, ax = plt.subplots()
 ax.set_xlim(1940, 2020)
-ax.set_ylim(0, max(engineers_data) + 100000)
+ax.set_ylim(0, max(widgets_data) + 100000)
 line, = ax.plot([], [])
 ax.set_xlabel('Decade')
-ax.set_ylabel('Number of Software Engineers')
+ax.set_ylabel('Number of Widgets')
 
 # Text annotation to display the current decade
 text = ax.text(0.5, 0.85, '', transform=ax.transAxes, fontsize=12, ha='center', va='center')
 
-def update(frame, line, text, decades, engineers_data):
+def update(frame, line, text, decades, widgets_data):
     """
     Update function for FuncAnimation.
 
@@ -36,17 +36,17 @@ def update(frame, line, text, decades, engineers_data):
         line (Line2D): The line object to update.
         text (Text): The text annotation object to update.
         decades (numpy.ndarray): Array of decades.
-        engineers_data (numpy.ndarray): Array of software engineers' data.
+        widgets_data (numpy.ndarray): Array of widgets' data.
 
     Returns:
         tuple: Tuple containing the updated Line2D and Text objects.
     """
     current_decade = decades[frame]
-    current_engineers = int(engineers_data[frame])
+    current_widgets = int(widgets_data[frame])
 
-    line.set_data(decades[:frame + 1], engineers_data[:frame + 1])
+    line.set_data(decades[:frame + 1], widgets_data[:frame + 1])
 
-    text.set_text(f'Decade: {current_decade}\nEngineers: {current_engineers}')
+    text.set_text(f'Decade: {current_decade}\nNumber of Widgets: {current_widgets}')
 
     return line, text
 
@@ -55,12 +55,9 @@ def update(frame, line, text, decades, engineers_data):
     fig,                    # Figure to update
     update,                 # Update function
     frames=len(decades),    # Number of frames
-    fargs=(line, text, decades, engineers_data),  # Additional arguments for the update function
+    fargs=(line, text, decades, widgets_data),  # Additional arguments for the update function
     interval=1000,          # Delay between frames in milliseconds
     blit=False,             # Whether to use blit for faster updates
 )
 
 plt.show()
-
-
-
