Split Dataframe into multiple dataframes based on if row contains string

Question

I want to split a DataFrame based on if a row contains a certain string "Customer" and all subsequent rows until another row that contains "Customer", i.e.

c_name       |   cost   | p_date
Customer: xx |          | 
hat          |    12    |   1/1/2018
bat          |    13    |   2/3/2018
Customer: xy |          |
mat          |    14    |   1/2/2018

to DF1

c_name       |   cost   | p_date
Customer: xx |          | 
hat          |    12    |   1/1/2018
bat          |    13    |   2/3/2018

DF2

c_name       |   cost   | p_date
Customer: xy |          |
mat          |    14    |   1/2/2018

I know I can get where to split the DataFrame based on df.c_name.str.contains('Customer'), but how do I use that to split the DataFrame?

user3483203 · Accepted Answer · 2018-12-06 22:23:41Z

5

`str.contains` + `groupby`

m = df.c_name.str.contains('Customer').cumsum()
d = {f'df{i}': g for i, g in df.groupby(m)}

for k, v in d.items():
    print(k)
    print(v, end='\n\n')

df1
         c_name  cost    p_date
0  Customer: xx   NaN       NaN
1           hat  12.0  1/1/2018
2           bat  13.0  2/3/2018

df2
         c_name  cost    p_date
3  Customer: xy   NaN       NaN
4           mat  14.0  1/2/2018

To access the individual DataFrames, simply use d['df1'], etc...

answered Dec 6, 2018 at 22:23

user3483203

51.3k10 gold badges72 silver badges104 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Split Dataframe into multiple dataframes based on if row contains string

1 Answer 1

`str.contains` + `groupby`

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

str.contains + groupby

Comments

Your Answer

Sign up or log in

Post as a guest

Related

`str.contains` + `groupby`