Панды: Создать счетчик событий для временного ряда, основанного на нескольких столбцах

Question

Мне нужно создать новую серию на моем панде DataFrame, чтобы подсчитать количество переключателей другого значения.Вот пример:

Мне нужна функция для вычисления столбца new_counter.DataFrame сортируется по foo и date.Счетчик увеличивается при изменении переменной foo или переменной baz.

Answer 1

Создайте помощника Series для последовательного счетчика и затем передайте его groupby с GroupBy.ngroup:

s = df['baz'].ne(df['baz'].shift()).cumsum()
df['counter'] = df.groupby(['foo', s]).ngroup() + 1