Mehrere Dataframes (Segmente) mit einer binären Maske erhalten

Question

Ich würde gerne wissen, ob es möglich ist (und wenn es so ist), mehrere Dataframes aus einer binären Maske mit Pandas zu erhalten.

Stellen Sie sich vor, dass ich die folgende Datenrahmen:

A B Mask 
0 0 1 0 
1 2 3 1 
2 4 5 1 
3 6 7 0 
4 8 9 1 
5 5 6 1 

ich ein dafaframe df1 erhalten möchten:

A B Mask 
1 2 3 1 
2 4 5 1 

und andere df2

A B Mask 
1 8 9 1 
2 5 6 1 

Ist dies möglich, ohne wiederkehrende Filter?

Answer 1

Eine Option:

[g[g.Mask != 0] for k, g in df.groupby((df.Mask == 0).cumsum()) if len(g) > 1] 

#[ A B Mask 
# 1 2 3  1 
# 2 4 5  1, 
# A B Mask 
# 4 8 9  1 
# 5 5 6  1]