python - Pandas 按范围分组，允许重复

Question

假设我有一个如下所示的数据框:

char  |  num
-----------------
A     |  1
B     |  2
C     |  3
D     |  4
E     |  5

有没有办法创建一个 groupby 对象，该对象将使用 3 范围内的“num”来创建如下所示的组:

第 1 组:

char  |  num
-----------------
A     |  1
B     |  2
C     |  3

第 2 组:

char  |  num
-----------------
B     |  2
C     |  3
D     |  4

第 3 组:

char  |  num
-----------------
C     |  3
D     |  4
E     |  5

行显然会重复，但这就是我想要的结果。我可以想出一些蛮力的方法来实现这一点，但我正在寻找一个优雅的解决方案(如果有的话)。

Answer 1

pd.concat 诡计

n = len(df) - 2
g = pd.concat([df.iloc[i: i + 3] for i in range(n)]).groupby(np.arange(n).repeat(3))

g.get_group(0)

  char  num
0    A    1
1    B    2
2    C    3