panda conteggio della stringa di conteggio del gruppo su colonna

Question

Desidero contare l'occorrenza di una stringa in una colonna di dataframe panda raggruppata.

Si supponga Ho il seguente dataframe:

catA catB scores 
A  X  6-4 RET 
A  X  6-4 6-4 
A  Y  6-3 RET 
B  Z  6-0 RET 
B  Z  6-1 RET 

In primo luogo, voglio gruppo da catA e catB. E per ciascuno di questi gruppi voglio contare l'occorrenza di RET nella colonna scores.

Il risultato dovrebbe essere qualcosa di simile:

catA catB RET 
A  X  1 
A  Y  1 
B  Z  2 

Il raggruppamento da due colonne è facile: grouped = df.groupby(['catA', 'catB'])

Ma che cosa è il prossimo?

Answer 1

chiamata apply su colonna i 'punti' sull'oggetto groupby e utilizzare il vectorise str metodo contains, utilizzare questo per filtrare il group e chiamare count:

In [34]:  
df.groupby(['catA', 'catB'])['scores'].apply(lambda x: x[x.str.contains('RET')].count()) 

Out[34]: 
catA catB 
A  X  1 
     Y  1 
B  Z  2 
Name: scores, dtype: int64 

Per assegnare un uso colonna transform affinché l'aggregazione restituisce una serie con il suo indice allineato al df originale:

In [35]: 
df['count'] = df.groupby(['catA', 'catB'])['scores'].transform(lambda x: x[x.str.contains('RET')].count()) 
df 

Out[35]: 
    catA catB scores count 
0 A X 6-4 RET  1 
1 A X 6-4 6-4  1 
2 A Y 6-3 RET  1 
3 B Z 6-0 RET  2 
4 B Z 6-1 RET  2