python-3.x - method - pandas plot set size

Número de pandas filas dentro del grupo

Question

Dado el siguiente marco de datos:

import pandas as pd
import numpy as np
df=pd.DataFrame({''A'':[''A'',''A'',''A'',''B'',''B'',''B''],
''B'':[''a'',''a'',''b'',''a'',''a'',''a''],
})
df
A B
0 A a
1 A a
2 A b
3 B a
4 B a
5 B a

Me gustaría crear la columna ''C'', que enumera las filas dentro de cada grupo en las columnas A y B de esta manera:

 A B C
0 A a 1
1 A a 2
2 A b 1
3 B a 1
4 B a 2
5 B a 3

He intentado esto hasta ahora:

df[''C'']=df.groupby([''A'',''B''])[''B''].transform(''rank'')

... pero no hay dados! ¡Gracias por adelantado!

Answer 1

Use groupby/cumcount :

In [25]: df[''C''] = df.groupby([''A'',''B'']).cumcount()+1; df
Out[25]:
A B C
0 A a 1
1 A a 2
2 A b 1
3 B a 1
4 B a 2
5 B a 3