GroupBy Pandas Подсчитайте последовательные нули ⇐ Python

[Anonymous]

Мой ввод выглядит так, как показано ниже.

Мне нужно сгруппировать по столбцу (A, B) и подсчитать количество последовательных нулей/подсчитать длину последовательные нули в каждой из групп и записываем в новый столбец «Zero_count»

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

Input:
A    B  DATE      hour  measure
A10  1  1/1/2014    0   0
A10  1  1/1/2014    1   0
A10  1  1/1/2014    2   0
A10  1  1/1/2014    3   0
A10  2  1/1/2014    4   0
A10  2  1/1/2014    5   1
A10  2  1/1/2014    6   2
A10  3  1/1/2014    7   0
A11  1  1/1/2014    8   0
A11  1  1/1/2014    9   0
A11  1  1/1/2014    10  2
A11  1  1/1/2014    11  0
A11  1  1/1/2014    12  0
A12  2  1/1/2014    13  1
A12  2  1/1/2014    14  3
A12  2  1/1/2014    15  0
A12  4  1/1/2014    16  5
A12  4  1/1/2014    17  0
A12  6  1/1/2014    18  0

Я пробовал использовать технику «groupby», чтобы получить группы, но мне нужен последовательный подсчет нулей внутри группы. Я пытался использовать лямбда-функцию, но она подсчитывает общее количество нулей, а меня интересует повторение последовательных нулей. Я хочу, чтобы мой результат выглядел так:

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

Output
A    B  DATE      hour  measure Consec_zero_count
A10  1  1/1/2014    0   0       4
A10  1  1/1/2014    1   0       4
A10  1  1/1/2014    2   0       4
A10  1  1/1/2014    3   0       4
A10  2  1/1/2014    4   0       1
A10  2  1/1/2014    5   1       0
A10  2  1/1/2014    6   2       0
A10  3  1/1/2014    7   0       1
A11  1  1/1/2014    8   0       2
A11  1  1/1/2014    9   0       2
A11  1  1/1/2014    10  2       0
A11  1  1/1/2014    11  0       2
A11  1  1/1/2014    12  0       2
A12  2  1/1/2014    13  1       0
A12  2  1/1/2014    14  3       0
A12  2  1/1/2014    15  0       1
A12  4  1/1/2014    16  5       0
A12  4  1/1/2014    17  0       1
A12  6  1/1/2014    18  0       1

Будем признательны за любые предложения. Заранее спасибо!

Подробнее здесь: https://stackoverflow.com/questions/527 ... tive-zeros