Как в Pandas определить «хитрую» оконную функцию?

Question

[Иван Мельников]

Имеется DataFrame df:

Нужно подсчитать для каждой строки длину последующей серии однотипных событий, причем считать нужно для каждого phone_number отдельно (а-ля, PARTITION BY phone_number ORDER BY timestamp). Нужный результат указан красным на скрине.
Ума не приложу как это сделать...

Answer 1

[Andrey Dugin]

У тебя на рисунке в одной строке ошибка

df['new_column'] = df.groupby(['phonenumber', 'event']).cumcount(ascending=False)

Comments

[Иван Мельников]

Проверил, у меня нет ошибки. В 3-й строке должна быть 1, так как после этого события есть ещё событие...

[Andrey Dugin]

Иван Мельников, нет, комбинация 8-456/busy встречается в таблице один раз и ниже этой строки её больше нет.

[Иван Мельников]

Andrey Dugin,

Нужно подсчитать для каждой строки длину последующей серии однотипных событий

Длина серии однотипных последующих событий, то есть равных между собой, но не обязательно равных событию из текущей строки...

[Andrey Dugin]

Иван Мельников, тогда переформулируй задачу понятным образом. Что есть "серия"? Она должна быть непрерывна для строк (т.е. подряд идущие строки) или непрерывна для номера (один и тот же номер и event, но строки могут перемежаться со строками от других номеров)?

[Иван Мельников]

Andrey Dugin, серия - это подряд идущие события одного типа (для данного phone_number), начиная со следующего от текущей строки.