在pandas中如何每隔n行分组并求和但保持原始结构

import pandas as pd
import numpy as np

# 创建一个示例DataFrame
df = pd.DataFrame({'A': range(10), 'B': range(10, 20)})

# 每隔3行分组并求和
n = 3

# 生成分组索引
groups = np.arange(len(df)) // n

# 按分组索引进行分组并求和
result = df.groupby(groups).sum()

# 将结果与原始DataFrame的结构保持一致
result = df.merge(result, left_index=True, right_index=True)

print(result)

   A   B  A_y  B_y
0  0  10    3   33
1  1  11    3   33
2  2  12    3   33
3  3  13   12  102
4  4  14   12  102
5  5  15   12  102
6  6  16   21  171
7  7  17   21  171
8  8  18   21  171
9  9  19    9   19