Python：如何使用列值復制 Dataframe 中的行，但將列值更改為其范圍

Question

有一個 dataframe df

Store   Aisle   Table
11      59      2
11      61      3

需要復制這些行 w.r.t。 更改“表”列值時的“表”列次數如下：

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

嘗試了下面的代碼，但這不會改變值，而是復制同一行 n 次。

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

謝謝！

Answer 1

之后您可以執行groupby().cumcount() ：

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

Answer 2

我們可以嘗試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