discard oregonator transient

dynamicslab · Jun 2, 2023 · 5f224ab · 5f224ab
1 parent 6ddfaa5
commit 5f224ab
Show file tree

Hide file tree

Showing 2 changed files with 78 additions and 86 deletions.
diff --git a/examples/17_parameterized_pattern_formation/parameterized_pattern_formation.ipynb b/examples/17_parameterized_pattern_formation/parameterized_pattern_formation.ipynb
diff --git a/examples/17_parameterized_pattern_formation/utils.py b/examples/17_parameterized_pattern_formation/utils.py
@@ -1020,11 +1020,13 @@ def generate_oregonator_trajectories():
                 Nt = X.shape[2] // 5 // 10 * 10
                 X0 = np.mean(X[:, :, -Nt:], axis=(0, 1, 2))
                 Z0 = np.mean(Z[:, :, -Nt:], axis=(0, 1, 2))
-                nt = X.shape[2] // 10 * 10
+                nt = X[:, :, Nt:].shape[2] // 10 * 10
                 A = np.mean(
                     (
-                        ((X - X0) / X0 + 1j * (Z - Z0) / Z0)
-                        * np.exp(-1j * 2 * np.pi / 10 * np.arange(X.shape[2]))
+                        ((X[:, :, Nt:] - X0) / X0 + 1j * (Z[:, :, Nt:] - Z0) / Z0)
+                        * np.exp(
+                            -1j * 2 * np.pi / 10 * np.arange(X[:, :, Nt:].shape[2])
+                        )
                     )[:, :, :nt].reshape(n, n, nt // 10, 10),
                     axis=3,
                 )