INNER CODE UNIT · Python
compute_probability_density
ssebastianmag/hydrogen-wavefunctions · hydrogen_wavefunction.py:173
def compute_probability_density(psi: np.ndarray):
""" Retrieve probability density |psi|^2.
Parameters:
psi (np.ndarray): Complex-valued coordinate-space wavefunction samples psi(x,z) on y=0.
Returns:
P (np.ndarray): Real-valued |psi|^2 with the same shape as psi.
"""
return np.abs(psi) ** 2
def compute_radial_probability_distribution(R: np.ndarray, r: np.ndarray):
""" Retrieve Radial probability distribution P_{n,l}(r) = r^2 * |R_{n,l}(r)|^2.
Parameters:
R (np.ndarray): Real-valued radial eigenfunction samples R_{n,l}(r).
r (np.ndarray): Radial coordinate(s) in meters.