Form D2_rho_0 from the matrix product D1@D1 for m=0 to incorporate Ne…

…umann boundary conditions.
HyQD · Dec 11, 2024 · 54c0f81 · 54c0f81
1 parent cde8a96
commit 54c0f81
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/notebooks/stationary_states_cylindrical_coordinates.ipynb b/notebooks/stationary_states_cylindrical_coordinates.ipynb
@@ -145,6 +145,7 @@
     "for j in range(1, N_rho):\n",
     "    G[0, j - 1] = -1 / D1[0, 0] * D1[0, j]\n",
     "D1_neumann = np.dot(D1, G)\n",
+    "D2_rho_0 = np.dot(D1, D1_neumann)[1:N_rho,:]\n",
     "D1_neumann = D1_neumann[1:N_rho, :]"
    ]
   },
@@ -176,7 +177,7 @@
     "\n",
     "#Setup the kinetic energy matrix for m=0\n",
     "T0 = -0.5 * (\n",
-    "    np.kron(D2_rho, I_z)\n",
+    "    np.kron(D2_rho_0, I_z)\n",
     "    + np.kron(I_rho, D2_z)\n",
     "    + np.kron(rho_inv_D1_rho_0, I_z)\n",
     ")\n",