Панды: группировка по столбцам и количество магазинов

Question

Как я могу создать новый столбец с именем «Count», чтобы для каждой строки он содержал количество зарегистрированных «Gender» соответствующей строки.Например, в приведенном ниже примере есть 2 зарегистрированных женщины и 1 зарегистрированный мужчина, поэтому для ввода:

Gender  Registered
female  1   
female  1
female  0
male    1
male    0
male    0

вывод должен быть:

Gender  Registered Count
female  1          2
female  1          2
female  0          2
male    1          1
male    0          1
male    0          1

Я пробовал что-то подобное, но это нене работает:

df[df['Registered'] == 1].groupby('Gender')['Gender'].count()

Answer 1

Использование transform

df.groupby('Gender').Registered.transform(lambda x : sum(x==1))
0    2
1    2
2    2
3    1
4    1
5    1
Name: Registered, dtype: int64
#df['count']=df.groupby('Gender').Registered.transform(lambda x : sum(x==1))