Comment la fonction « groupby » de Pandas peut-elle créer efficacement des listes de valeurs à partir de lignes groupées ?

Regrouper efficacement les lignes en listes dans Pandas via Groupby

Pour les tâches de manipulation de données, le regroupement des lignes en listes en fonction de critères spécifiques est une exigence courante. Dans Pandas, la fonction groupby fournit un outil puissant à cet effet.

Supposons que vous ayez un DataFrame avec deux colonnes, 'a' et 'b' :

a b
A 1
A 2
B 5
B 5
B 4
C 6

Le but est de regroupez les lignes en fonction de la colonne « a » et créez des listes de la colonne « b » pour chaque groupe.

Pour y parvenir, vous pouvez tirer parti du groupby function :

df.groupby('a')['b'].apply(list)

La fonction groupby regroupe le DataFrame par la colonne 'a'. La fonction apply parcourt ensuite chaque groupe et convertit la colonne « b » en liste à l'aide de list.

Le résultat résultant :

a
A       [1, 2]
B    [5, 5, 4]
C          [6]
Name: b, dtype: object

Cette technique vous permet de regrouper efficacement les lignes en fonction de une colonne spécifique et obtenir des listes de valeurs pour d'autres colonnes au sein de chaque groupe.

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!