Pandas -在分组后移动滚动和溢出到以下组

import pandas as pd

# 创建一个DataFrame
data = {'group': ['A', 'A', 'A', 'B', 'B', 'B'],
        'value': [1, 2, 3, 4, 5, 6]}
df = pd.DataFrame(data)

# 按照group列进行分组
grouped = df.groupby('group')

# 在每个组内进行滚动计算
rolling_sum = grouped['value'].rolling(window=2, min_periods=1).sum()

# 将滚动计算结果添加到原始DataFrame中
df['rolling_sum'] = rolling_sum.reset_index(level=0, drop=True)

# 打印结果
print(df)

  group  value  rolling_sum
0     A      1          1.0
1     A      2          3.0
2     A      3          5.0
3     B      4          4.0
4     B      5          9.0
5     B      6         11.0