code: def get_occupation(self, atom_index, orbital):
          
  
          orbital_index = self.orbitals.index(orbital)
          return {spin: np.sum(d[:, :, atom_index, orbital_index] * self.weights[:, None])
                  for spin, d in self.data.items()}
docs: Returns the occupation for a particular orbital of a particular atom.
  
          Args:
              atom_num (int): Index of atom in the PROCAR. It should be noted
                  that VASP uses 1-based indexing for atoms, but this is
                  converted to 0-based indexing in this parser to be
                  consistent with representation of structures in pymatgen.
              orbital (str): An orbital. If it is a single character, e.g., s,
                  p, d or f, the sum of all s-type, p-type, d-type or f-type
                  orbitals occupations are returned respectively. If it is a
                  specific orbital, e.g., px, dxy, etc., only the occupation
                  of that orbital is returned.
  
          Returns:
              Sum occupation of orbital of atom.