Comment regrouper des données par colonne et les convertir en dictionnaire de listes en Python ?

Résultats GroupBy vers dictionnaire de listes

Vous disposez d'une feuille de calcul Excel avec trois colonnes, Colonne1, Colonne2 et Colonne3. Vous souhaitez extraire les données, les regrouper par Colonne1, et créer un dictionnaire de listes qui ressemble à ceci :

{0: [1],
1: [2, 3, 5],
2: [1, 2],
3: [4, 5],
4: [1],
5: [1, 2, 3]}

Pour arriver à ce résultat, vous pouvez utiliser le code suivant :

excel = pandas.read_excel(r"e:\test_data.xlsx", sheetname='mySheet', parse_cols="A,C")
grouped_data = excel.groupby("Column1")["Column3"].apply(list).to_dict()

La fonction groupby regroupe les données par Colonne1 et renvoie un objet groupby. La méthode apply est ensuite utilisée pour appliquer la fonction de liste aux données Column3, créant ainsi une liste de valeurs pour chaque groupe. Enfin, la méthode to_dict est utilisée pour convertir les données groupées en un dictionnaire de listes.

Le dictionnaire résultant aura le format que vous avez spécifié, les clés étant les valeurs de la colonne 1 et les valeurs étant les listes de la valeurs correspondantes dans la colonne 3.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!