examples: animations numpy2.x related changes

Author: fs446

doc/examples/animations-pulsating-sphere.ipynb

@@ -147,7 +147,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "grid = hex_grid([xmin, xmax], [ymin, ymax], 0.0125, 'vertical')\n",
+    "grid = sfs.util.as_xyz_components(hex_grid([xmin, xmax],\n",
+    "                                           [ymin, ymax],\n",
+    "                                           0.0125,\n",
+    "                                           'vertical'))\n",
     "ani = animation.particle_displacement(\n",
     "        omega, center, radius, amplitude, grid, frames, figsize, c='Gray')\n",
     "plt.close()\n",
@@ -167,8 +170,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "grid = [np.random.uniform(xmin, xmax, 4000),\n",
-    "        np.random.uniform(ymin, ymax, 4000), 0]\n",
+    "grid = sfs.util.as_xyz_components([np.random.uniform(xmin, xmax, 4000),\n",
+    "                                   np.random.uniform(ymin, ymax, 4000),\n",
+    "                                   0])\n",
     "ani = animation.particle_displacement(\n",
     "        omega, center, radius, amplitude, grid, frames, figsize, c='Gray')\n",
     "plt.close()\n",

doc/examples/animations_pulsating_sphere.py

@@ -3,6 +3,8 @@
 import numpy as np
 from matplotlib import pyplot as plt
 from matplotlib import animation
+import warnings
+warnings.simplefilter("ignore", np.exceptions.ComplexWarning)
 
 
 def particle_displacement(omega, center, radius, amplitude, grid, frames,
@@ -18,7 +20,7 @@ def particle_displacement(omega, center, radius, amplitude, grid, frames,
     scat = sfs.plot2d.particles(grid + displacement, **kwargs)
 
     def update_frame_displacement(i):
-        position = (grid + displacement * phasor**i).apply(np.real)
+        position = np.real((grid + displacement * phasor**i))
         position = np.column_stack([position[0].flatten(),
                                     position[1].flatten()])
         scat.set_offsets(position)
@@ -43,7 +45,7 @@ def particle_velocity(omega, center, radius, amplitude, grid, frames,
             **kwargs)
 
     def update_frame_velocity(i):
-        quiv.set_UVC(*(velocity[:2] * phasor**i).apply(np.real))
+        np.real(quiv.set_UVC(*(velocity[:2] * phasor**i)))
         return [quiv]
 
     return animation.FuncAnimation(

doc/examples/plot_particle_density.py

@@ -14,7 +14,9 @@
 # normal vector of plane wave
 npw = sfs.util.direction_vector(np.radians(pw_angle))
 # random grid for velocity
-grid = [np.random.uniform(-3, 3, 40000), np.random.uniform(-3, 3, 40000), 0]
+grid = sfs.util.as_xyz_components([np.random.uniform(-3, 3, 40000),
+                                   np.random.uniform(-3, 3, 40000),
+                                   0])
 
 
 def plot_particle_displacement(title):