Панды добавляют последнюю строку предыдущей группы ⇐ Python

[Anonymous]

У меня есть следующий столбец B, который я хотел бы использовать для получения столбцов C и D (где B + C = D):

Код: Выделить всё

B       C        D

1       0        1
2       0        2
3       0        3
0       3        3
0       3        3
0       3        3
1       3        4
2       3        5
3       3        6
0       6        6
1       6        7

К сожалению, у меня возникли проблемы с поиском наилучшего способа сделать это.
Я пробовал использовать различные комбинации .groupby (), прокрутка Apply, .ffill() и .cumsum() безрезультатно.
Мне удалось разделиться Б с использованием

Код: Выделить всё

x = (df['B'].ne(0).diff() & df['B'].shift().ne())

но в остальном я в тупике.
Буду признателен за любую помощь!

Подробнее здесь: https://stackoverflow.com/questions/785 ... ious-group