updated passing_args_guide

vamshipy · vamshipy · commit ffac8a1f75ba · 2024-01-23T09:21:22.000+05:30
diff --git a/galleries/examples/animation/QuadMesh.get_cursor_data().py b/galleries/examples/animation/QuadMesh.get_cursor_data().py
@@ -0,0 +1,39 @@
+import matplotlib.pyplot as plt
+import numpy as np
+
+plt.rcParams["backend"] = "TKAgg"
+plt.ion()
+fig, ax = plt.subplots()
+
+data = np.arange(6).reshape(2, 3)
+
+qm = ax.pcolormesh(
+    ["col 1", "col 2", "col 3"],
+    ["row 1", "row 2"],
+    data,
+)
+
+
+def on_mouse_move(event):
+    if not event.inaxes:
+        return
+    cd = qm.get_cursor_data(event)
+    if cd is None:
+        return
+
+    # The value of the cell, according to get_cursor_data()
+    cd_val = cd[0]
+
+    # The value of the cell, according to the data row/col
+    data_val = data[round(event.ydata), round(event.xdata)]
+
+    if cd_val == data_val:
+        print("Fine")
+    else:
+        print(event)
+        print(cd)
+        raise ValueError(f"Value mismatch. {cd_val} != {data_val}")
+
+
+fig.canvas.mpl_connect("motion_notify_event", on_mouse_move)
+fig.canvas.manager.window.geometry("800x1054+100+100")
diff --git a/galleries/examples/animation/animated_FuncAnimation.py b/galleries/examples/animation/animated_FuncAnimation.py
@@ -1,38 +1,18 @@
 """
-==================
+==================================
 Parameterized Animation Functions
-==================
+==================================
 use FuncAnimation to show counts of widgets over the decades
 """
+import functools
+
 import matplotlib.pyplot as plt
 import numpy as np
 
 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):
-    # Update function for FuncAnimation.
-
     # Parameters:
     # frame (int): The current frame number.
     # line (Line2D): The line object to update.
@@ -53,11 +33,32 @@ 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
-    update,                 # Update function
+    functools.partial(update,  line=line, text=text,
+                      decades=decades, widgets_data=widgets_data),
     frames=len(decades),    # Number of frames
-    fargs=(line, text, decades, widgets_data),  # Additional arguments for update
     interval=1000,          # Delay between frames in milliseconds
     blit=False,             # Whether to use blit for faster updates
 )
