Подсчитать количество нулей между двумя ненулевыми числами последовательно в кадре данных

Ответ 1 · рейтинг 3 · jezrael

Для повышения производительности можно использовать векторизованное решение, это похоже на это решение с Series.shift столбца и сравнить по 0:

a = df['value'].shift().eq(0)
b = a.cumsum()
df['new'] = b.sub(b.mask(a).ffill().fillna(0)).astype(int)
print (df)
   value  zeroCumulative  new
0     70               0    0
1    127               0    0
2    101               0    0
3      0               0    0
4      0               1    1
5      0               2    2
6     73               3    3
7      0               0    0
8     55               1    1
9      0               0    0