python - 用计数复制数据框中的每一行

Question

对于数据框中的每一行，我希望创建它的副本，并添加一个列来标识每个副本。

例如原始数据框是

A | A 

B | B 

我希望复制每一行并增加一列来标识它。结果:

A | A | 1

A | A | 2

B | B | 1

B | B | 2

Answer 1

您可以使用 df.reindex 后跟 groupby 在 df.index .

df = df.reindex(df.index.repeat(2))
df['count'] = df.groupby(level=0).cumcount() + 1
df = df.reset_index(drop=True)

df
   a  b  count
0  A  A      1
1  A  A      2
2  B  B      1
3  B  B      2

---

同样，使用 reindex 和 assign 与 np.tile :

df = df.reindex(df.index.repeat(2))\
       .assign(count=np.tile(df.index, 2) + 1)\
       .reset_index(drop=True)

df

   a  b  count
0  A  A      1
1  A  A      2
2  B  B      1
3  B  B      2