def _dmet_orb_list(mol, atom_list):
    """Rearrange the orbital label

    Args:
        mol (pyscf.gto.Mole): The molecule to simulate.
        atom_list (list): Atom list for IAO assignment (int).

    Returns:
        newlist (list): The orbital list in new order (int).
    """
    newlist = []
    for i in range(mol.natm):
        for j in range(mol.nao_nr()):
            if (atom_list[j] == i):
                newlist.append(j)

    return newlist