在Pandas dataframe中使用groupby时，如何根据前一行中的值递增一列？

import pandas as pd

# 创建示例数据
data = {'A': ['a', 'a', 'b', 'b', 'b'],
        'B': [1, 2, 3, 4, 5]}

df = pd.DataFrame(data)

# 根据列'A'进行分组
grouped = df.groupby('A')

# 使用shift()函数获取前一行的值
df['C'] = grouped['B'].shift()

# 使用cumsum()函数计算递增的累加和
df['C'] = df.groupby('A')['C'].cumsum()

print(df)

   A  B    C
0  a  1  NaN
1  a  2  1.0
2  b  3  NaN
3  b  4  3.0
4  b  5  7.0