Вычислить дополнительное значение на основе другого столбца ⇐ Python

[Anonymous]

Имея такой фрейм данных pandas

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

df = pd.DataFrame({
'ts_diff':[0, 0, 738, 20, 29, 61, 42, 18, 62, 41, 42, 0, 0, 729, 43, 59, 42, 61, 44, 36, 61, 61, 42, 18, 62, 41, 42, 0, 0]
})

ts_diff — продолжительность между событиями в миллисекундах.
Хотелось бы создать другой столбец ts_diff_incr, основанный на ts_diff.

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

ts_diff_incr

 — продолжительность в миллисекундах между текущим шагом и началом события (где ts_diff равен нулю).
Пожалуйста, посмотрите логику вычислений на изображении ниже.
Как лучше всего добиться этого без циклов for и вместо этого использовать векторизацию?



Подробнее здесь: https://stackoverflow.com/questions/798 ... her-column