Панды DataFrame: Как GroupBy последовательных значений

Question

У меня есть столбец в DataFrame со значениями:

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

Как я могу группировать их, как это?

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

Answer 1

Вы можете использовать groupby обычаем 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]