INNER CODE UNIT · Python

psi

ssebastianmag/hydrogen-wavefunctions · hydrogen_wavefunction.py:168

    psi = R * Y

    return Xg, Zg, psi, a_mu


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):

View source record →