Update animated_FuncAnimation.py

vamshipy · web-flow · commit c8e34deca16c · 2024-01-23T09:04:50.000+05:30
diff --git a/galleries/examples/animation/animated_FuncAnimation.py b/galleries/examples/animation/animated_FuncAnimation.py
@@ -9,26 +9,6 @@
 
 import matplotlib.animation as animation
 
-# Constants
-decades = np.arange(1940, 2020, 10)
-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))
-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(widgets_data) + 100000)
-line, = ax.plot([], [])
-ax.set_xlabel('Decade')
-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, widgets_data):
     # Parameters:
@@ -51,6 +31,27 @@ def update(frame, line, text, decades, widgets_data):
     return line, text
 
 # Set up the animation
+
+# Constants
+decades = np.arange(1940, 2020, 10)
+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))
+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(widgets_data) + 100000)
+line, = ax.plot([], [])
+ax.set_xlabel('Decade')
+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')
+
 ani = animation.FuncAnimation(
     fig,                    # Figure to update
     functools.partial(update,  line=line, text=text, decades=decades, widgets_data=widgets_data),
