Python pandy: Dodaj kolumnę do mojego dataframe że liczy zmienny

Question

Mam dataframe „GT” tak:

org  group 
org1  1 
org2  1 
org3  2 
org4  3 
org5  3 
org6  3 

i chciałbym dodać kolumny „count” GT dataframe się liczy członka numer grup, spodziewane wyniki, takie jak ten:

org  group count 
org1  1  2 
org2  1  2 
org3  2  1 
org4  3  3 
org5  3  3 
org6  3  3 

wiem jak to zrobić na jednej pozycji w grupie, ale nie wiem jak zrobić licznik powtarzany dla wszystkich elementów grupy, oto kod Użyłem:

gtcounts = gt.groupby('group').count() 

Czy ktoś może pomóc?

Answer 1

połączeń transform to zwróci Series wyrównany z oryginalnym DF:

In [223]: 

df['count'] = df.groupby('group')['group'].transform('count') 
df 
Out[223]: 
    org group count 
0 org1  1  2 
1 org2  1  2 
2 org3  2  1 
3 org4  3  3 
4 org5  3  3 
5 org6  3  3