Совокупное количество, если идентификатор в предыдущей строке совпадает - Python / Pandas

Question

Это то, чего я пытаюсь достичь в python. В Excel есть простая формула, каков питонный способ справиться с этим?

CUM_FREQ      ID      

1          100001
2          100001
3          100001
4          100001
1          100002
2          100002
1          100003
2          100003
3          100003
4          100003

Каждый раз при изменении идентификатора CUM_FREQ строка должна быть сброшена в 1. Excel Formula IF (C3 = C2, B2 + 1,1)

Заранее спасибо! Рахул

Answer 1

Одним простым способом было бы иметь идентификаторы в списке и использовать Счетчик для вычисления количества.

from collections import Counter

id_list = [100001, 100001, 100001, 100001, 100002, 100002, 100003, 100003, 100003, 100003]
cum_freq = []
for i, v in enumerate(id_list):
    cum_freq.append(Counter(id_list[:i+1])[v])

print(cum_freq)
[1, 2, 3, 4, 1, 2, 1, 2, 3, 4]