如何扩展 pandas 数据框，使每一行都成为之前的所有行？

Question

我的目标是能够像我在下面手动完成的那样为数据框中的每个组累积添加行，但不使用 for 循环或 df.apply()（所以基本上是一个操作）。

import pandas as pd
import numpy as np

df1 = pd.DataFrame(np.array([[1, 1, 1], [2, 2, 2], [3, 3, 3]]),
                   columns=['group', 'a', 'b'])

df2 = pd.DataFrame(np.array([[1, 1, 1], [2, 1, 1], [2, 2, 2], [3, 1, 1], [3, 1, 1], [3, 2, 2]]),
                   columns=['group', 'a', 'b'])

df1 = df1.set_index('group').sort_index()
df2 = df2.set_index('group').sort_index()

print(df1)

       a  b
group      
1      1  1
2      2  2
3      3  3

print(df2)

       a  b
group      
1      1  1
2      1  1
2      2  2
3      1  1
3      1  1
3      2  2

Answer 1

IIUC，你可以使用：

tmp = pd.DataFrame(1, columns=df1.columns, index=df1.index.repeat(range(len(df1))))
df2 = pd.concat([tmp, df1]).sort_index()
print(df2)

# Output
       a  b
group      
1      1  1
2      1  1
2      2  2
3      1  1
3      1  1
3      3  3

一行：

df2 = pd.concat([pd.DataFrame(1, columns=df1.columns, index=df1.index.repeat(range(len(df1)))), df1]).sort_index()

如何扩展 pandas 数据框，使每一行都成为之前的所有行？

How do I expand a pandas data frame such that each row becomes all previous rows?

python

pandas

dataframe