python - 要插入数据框-pandas 中的选定行

Question

如何在 pandas 数据框中插入行？我有一个大数据框，我正在尝试识别要重复行并插入数据框中的特定值。例如:

df1 = pd.DataFrame([[1, 2], [3, 4],[1, 6],[2,3],[1,5]], columns=["a", "b"])
   a  b
0  1  2
1  3  4
2  1  6
3  2  3
4  1  5

列“a”代表要在数据框中插入的行重复次数，所以我想得到以下结果

   a  b
0  1  2
1  1  4
2  1  4
3  1  4
4  1  6
5  1  3
6  1  3
7  1  5

我尝试使用追加，但结果不是我所期望的。这是我到目前为止所拥有的。我将不胜感激任何见解。

df2 = df1[df1.a > 1]               # To select rows with values more than 1
repeats = (df2.iloc[0]["a"] - 1)   # number of repetitions -1
r2 = pd.concat([df2]*repeats, ignore_index=True)
df_modified = df1.append(r2, ignore_index=True)

Answer 1

使用重新索引和重复

df1.reindex(df1.index.repeat(df1.a)).assign(a=1).reset_index(drop=True)
Out[1266]: 
   a  b
0  1  2
1  1  4
2  1  4
3  1  4
4  1  6
5  1  3
6  1  3
7  1  5