INNER CODE UNIT · Python
compute_radial_probability_distribution
ssebastianmag/hydrogen-wavefunctions · hydrogen_wavefunction.py:185
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.
Returns:
P_r (np.ndarray): Real-valued P_{n,l}(r) with the same shape as r.
"""
return (r**2) * np.abs(R) ** 2
def reduced_electron_nucleus_mass(Z: int, M: Optional[float] = None):
""" Compute electron–nucleus reduced mass μ.
Parameters:
Z (int): Nuclear charge number. (Z=1 for Hydrogen, Z>1 for hydrogenic ions).