如何从pandas数据框架中的事务表中获取唯一ID的月计数

Question

ID  month   amnt    flg
123 Jan       10    0
123 feb       20    1
123 mar       30    1
124 jan       15    1
124 feb       15    1
124 mar       20    1
125 feb       20       1

我有上面的数据框。这面旗帜是为拥有超过1.5万美元的客户创建的。数据有3个月的数据，所以创建下表的3个月。我希望在

下面

ID      Months
123        3
124         3
125         1

我尝试为每个月创建标志，然后尝试按我和月份分组。但这并没有奏效。我试着

df.groupby(['ID'])['month'].count()

需要你的帮助。

Answer 1

try:

li = [1,2,3]
x = df.groupby(['ID'])['month'].apply(lambda x: x.count()).values
y = {f'month{p}of3':sum(x == p) for p in li}
df = pd.DataFrame([y])

---

df:

month1of3   month2of3   month3of3
0   1           0           2