Pandas groupby Anzahl Zeichenfolge Auftreten über Spalte

Question

Ich möchte das Auftreten einer Zeichenfolge in einer gruppierten Pandas Dataframe Spalte zählen.

Angenommen ich folgende Datenrahmen haben:

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 

Zuerst möchte ich zu einer Gruppe von catA und catB. Und für jede dieser Gruppen möchte ich das Auftreten von RET in der Spalte scores zählen.

Das Ergebnis sollte wie folgt aussehen:

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

Die Gruppierung von zwei Spalten ist einfach: grouped = df.groupby(['catA', 'catB'])

Aber was kommt als nächstes?

Answer 1

Anruf apply auf der Spalte ‚Noten‘ auf der groupby Objekt und verwenden die vectorize str Methode contains verwenden dies die group und rufen zu filtern 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 

als Spalten Verwendung zuzuweisen transform so dass Die Aggregation gibt eine Reihe mit ihrem Index zurück, der auf das ursprüngliche df ausgerichtet ist:

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