example for funcAnimation

vamshipy · vamshipy · commit a6de6a4ff4cd · 2024-01-19T21:58:56.000+05:30
diff --git a/galleries/examples/animation/animated_FuncAnimation.py b/galleries/examples/animation/animated_FuncAnimation.py
@@ -0,0 +1,68 @@
+"""
+==================
+Animated graph using funcAnimation
+==================
+use funcAnimation to create a graph of no of
+software engineers over the decades
+"""
+import matplotlib.pyplot as plt
+import matplotlib.animation as animation
+import numpy as np
+
+# Constants
+decades = np.arange(1940, 2020, 10)
+initial_engineers = 10000  # Rough estimate of the number of software engineers(not real) 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
+
+# Set up the initial plot
+fig, ax = plt.subplots()
+ax.set_xlim(1940, 2020)
+ax.set_ylim(0, max(engineers_data) + 100000)
+line, = ax.plot([], [], label='Software Engineers')
+ax.set_xlabel('Decade')
+ax.set_ylabel('Number of Software Engineers')
+ax.legend()
+
+# 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):
+    """
+    Update function for FuncAnimation.
+
+    Parameters:
+        frame (int): The current frame number.
+        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.
+
+    Returns:
+        tuple: Tuple containing the updated Line2D and Text objects.
+    """
+    current_decade = decades[frame]
+    current_engineers = int(engineers_data[frame])
+
+    line.set_data(decades[:frame + 1], engineers_data[:frame + 1])
+
+    text.set_text(f'Decade: {current_decade}\nEngineers: {current_engineers}')
+
+    return line, text
+
+# Set up the animation
+ani = animation.FuncAnimation(
+    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
+    interval=1000,          # Delay between frames in milliseconds
+    blit=False,             # Whether to use blit for faster updates
+)
+
+plt.show()
+
+
+
