Найдите следующую строку, соответствующую условию

#python #pandas

Вопрос:

У меня есть фрейм данных, который выглядит примерно так:

пользователь Тип Партнеры Дата
пользователь_а фут п_а 2021-01-01
пользователь_а КП 0 2021-01-02
пользователь_а КП p_b 2021-01-03
пользователь_а фут p_b 2021-01-04
пользователь_а КП п_а 2021-01-05
пользователь_а КП p_b 2021-01-06

Записи сортируются по пользователям и по дате. Для каждой строки с типом » ft » мне нужно количество строк с типом = cp между этой строкой и следующей с типом = cp И теми же партнерами И тем же пользователем. Мне также нужна дата этой записи.

С предыдущим образцом я хотел бы иметь :

пользователь Тип Партнеры Дата n_rows следующая дата
пользователь_а фут п_а 2021-01-01 2 2021-01-05
пользователь_а фут p_b 2021-01-04 1 2021-01-06

Есть ли способ сделать это в Python ?