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