python - 根据 cumsum 值对行求和

Question

我有一个数据框

指数 A B C
0 4 7 9
1 2 6 2
2 6 9 1
3 7 2 4
4 8 5 6

我想根据 C 列的总和创建另一个数据框。但这里的问题是，如果 C 的总和达到 10 或更高，它应该创建另一行。像这样。

指数 A B C
0 6 13 11
1 21 16 11

任何帮助都将不胜感激。是否有可靠的方法来执行此操作，或者迭代是我最后的选择？

Answer 1

有一种非迭代的方法。您需要一个基于 C % 11 的 groupby。

# Groupby logic - https://stackoverflow.com/a/45959831/4909087
out = df.groupby((df.C.cumsum() % 10).diff().shift().lt(0).cumsum(), as_index=0).agg('sum')
print(out) 
    A   B   C
0   6  13  11
1  21  16  11