python - 如何使用类似分组的函数将值插入新列？

Question

我想知道是否有一种方法可以根据使用类似于 df.groupby(['Col1','Col2']).agg({'Col2':'count'}) 的某种代码将值插入到数据帧的新列中功能。

我的 Df 看起来像这样:

ID Customer Connection Product_ID Count
10 AMXX     Instant    AAB        NaN 
11 JKXX     Slow       AAB        NaN
12 LKXX     Slow       HJA        NaN
13 AMXX     Instant    AAB        NaN
14 RFXX     Slow       WRQ        NaN
15 RFXX     Instant    WRQ        NaN

df['Count'] 是我创建的新空列，我想在其中存储每个客户使用“Product_ID”列中每行记录的产品的次数。我希望使用相同的 df 并且只填写“计数”列，而不是进行分组。

我希望 df 看起来像这样:

ID Customer Connection Product_ID Count
10 AMXX     Instant    AAB        2
11 JKXX     Slow       AAB        1
12 LKXX     Slow       HJA        1
13 AMXX     Instant    AAB        2
14 RFXX     Slow       WRQ        2
15 RFXX     Instant    WRQ        2

有人知道我怎么能做到这一点吗？ 谢谢:)

Answer 1

尝试:

df['Count'] = df.groupby('Customer')['Product_ID'].transform('count')