Automatiser la création de groupes de personnes

Bonjour,

Je dispose d'un dataframe qui représente des personnes, et leurs deux souhaits d'appartenance à un groupe de travail (il y a 5 groupes de travail, et ils doivent se départager. Pour cela, ils énoncent 2 souhaits). On a donc une première colonne de nom de personnes, une deuxième colonne (qui représente un premier vœux d'appartenance à un groupe) où ces personnes devaient entrer un numéro de groupe (de 1 à 5), et une troisième colonne (qui représente un second vœux d'appartenance à un groupe) où les personnes devaient entrer un numéro de groupe (de 1 à 5).

Voici un exemple reproductible :

import pandas as pd
import numpy as np
from numpy import random

data = pd.DataFrame({"prenom" : ["personne1", "personne2", "personne3", "personne4", "personne5", "personne6", "personne7", "personne8", "personne9", "personne10", "personne11", "personne12", "personne13", "personne14", "personne15", "personne16", "personne17", "personne18", "personne19", "personne20"], "choix1" :  np.random.randint(low=1, high=5, size=(20,)), "choix2" : np.random.randint(low=1, high=5, size=(20,))})

Je souhaite créer un programme qui répartissent aléatoirement ces personnes dans ces 5 groupes de travail (5 groupes de 4 personnes) en prenant en compte leurs choix. En priorité, on se base sur leur premier vœux ; sinon, on se base sur leur deuxième vœux. L'objectif, c'est que les 5 groupes possèdent bien 4 personnes.

Est-il possible de faire cela sur Python ?
Merci.