web-dev-qa-db-fra.com

Comment grouper des valeurs consécutives dans pandas DataFrame

J'ai une colonne dans un DataFrame avec des valeurs:

[1, 1, -1, 1, -1, -1]

Comment puis-je les regrouper comme ça?

[1,1] [-1] [1] [-1, -1]
17
Bryan Fok

Vous pouvez utiliser groupby par personnalisé Series:

df = pd.DataFrame({'a': [1, 1, -1, 1, -1, -1]})
print (df)
   a
0  1
1  1
2 -1
3  1
4 -1
5 -1

print ((df.a != df.a.shift()).cumsum())
0    1
1    1
2    2
3    3
4    4
5    4
Name: a, dtype: int32
for i, g in df.groupby([(df.a != df.a.shift()).cumsum()]):
    print (i)
    print (g)
    print (g.a.tolist())

   a
0  1
1  1
[1, 1]
2
   a
2 -1
[-1]
3
   a
3  1
[1]
4
   a
4 -1
5 -1
[-1, -1]
31
jezrael

Utilisation de groupby de itertools des données de Jez

from itertools import groupby
[ list(group) for key, group in groupby(df.a.values.tolist())]
Out[361]: [[1, 1], [-1], [1], [-1, -1]]
12
YOBEN_S