jaxrts.elements.electron_distribution

jaxrts.elements.electron_distribution(atomic_number: int) → Array[source]

Returns number of electrons for each orbital (defined by the quantum numbers n and l) for a neutral atom with the given atomic_number.

Parameter,

atomic_number: int: The atomic number of the element.

returns:: jnp.ndarray – The number of electrons per orbital as a flat array, starting with 1s, up to 4f. To find the corresponding indices for individual orbitals, one can use orbital_map