CODEWITHSUNDEEP
pythonpandasrepeat
user12345
user12345

Reputation: 509

Python: How to replicate rows in Dataframe with column value but changing the column value to its range

Have got a dataframe df

Store   Aisle   Table
11      59      2
11      61      3

Need to replicate these rows w.r.t. column 'Table' times on changing 'Table' column value as below:

Store   Aisle   Table
11      59      1
11      59      2
11      61      1
11      61      2
11      61      3

Tried below code, but this doesn't change the value instead replicates the same row n times.

df.loc[df.index.repeat(df['Table'])]

Thanks!

Upvotes: 1

Views: 207

Answers (2)

BENY
BENY

Reputation: 323266

We can try explode

out = df.assign(Table=df['Table'].map(range)).explode('Table')
Out[160]: 
   Store  Aisle Table
0     11     59     0
0     11     59     1
1     11     61     0
1     11     61     1
1     11     61     2

Upvotes: 1

Quang Hoang
Quang Hoang

Reputation: 150745

You can do a groupby().cumcount() after that:

out = df.loc[df.index.repeat(df['Table'])]
out['Table'] = out.groupby(level=0).cumcount() + 1

Output:

   Store  Aisle  Table
0     11     59      1
0     11     59      2
1     11     61      1
1     11     61      2
1     11     61      3

Upvotes: 2

Related Questions