Een groupby-som uitvoeren in pandas (met voorbeelden)
U kunt de volgende basissyntaxis gebruiken om de som van waarden per groep in panda’s te vinden:
df. groupby ([' group1 ',' group2 '])[' sum_col ']. sum (). reset_index ()
De volgende voorbeelden laten zien hoe u deze syntaxis in de praktijk kunt gebruiken met de volgende panda’s DataFrame:
import pandas as pd #createDataFrame df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'], ' position ': ['G', 'G', 'F', 'C', 'G', 'F', 'F', 'C'], ' points ': [25, 17, 14, 9, 12, 9, 6, 4], ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12]}) #view DataFrame df team position points rebounds 0 A G 25 11 1 A G 17 8 2 A F 14 10 3 A C 9 6 4 B G 12 6 5 B F 9 5 6 B F 6 9 7 B C 4 12
Voorbeeld 1: groeperen op een kolom, een kolom toevoegen
De volgende code laat zien hoe u op kolom kunt groeperen en de waarden in een kolom kunt optellen:
#group by team and sum the points
df. groupby ([' team '])[' points ']. sum (). reset_index ()
team points
0 to 65
1 B 31
Uit het resultaat kunnen we zien dat:
- Spelers van team A scoorden in totaal 65 punten.
- Spelers van team B scoorden in totaal 31 punten.
Voorbeeld 2: groeperen op meerdere kolommen, meerdere kolommen toevoegen
De volgende code laat zien hoe u over meerdere kolommen kunt groeperen en meerdere kolommen kunt optellen:
#group by team and position, sum points and rebounds
df. groupby ([' team ', ' position '])[' points ', ' rebounds ']. sum (). reset_index ()
team position points rebounds
0 A C 9 6
1 A F 14 10
2 A G 42 19
3 B C 4 12
4 B F 15 14
5 B G 12 6
Uit het resultaat kunnen we zien dat:
- Spelers van team A in positie “C” scoorden in totaal 9 punten en 6 rebounds.
- Spelers van team A in positie “F” scoorden in totaal 14 punten en 10 rebounds.
- Team A-spelers in de “G”-positie scoorden in totaal 42 punten en 19 rebounds.
Enzovoort.
Houd er rekening mee dat de functie reset_index() voorkomt dat het groeperen van kolommen deel uitmaakt van de index.
Zo ziet het resultaat er bijvoorbeeld uit als we het niet gebruiken:
#group by team and position, sum points and rebounds
df. groupby ([' team ', ' position '])[' points ', ' rebounds ']. sum ()
rebound points
team position
A C 9 6
F 14 10
G 42 19
B C 4 12
F 15 14
G 12 6
Afhankelijk van hoe u wilt dat de resultaten verschijnen, kunt u er wel of niet voor kiezen om de reset_index() functie te gebruiken.
Aanvullende bronnen
In de volgende tutorials wordt uitgelegd hoe u andere veelvoorkomende groeperingsbewerkingen in panda’s kunt uitvoeren:
Hoe groepswaarnemingen bij panda’s te tellen
Zo vind je bij Pandas de maximale waarde per groep
Hoe kwantielen per groep in Panda’s te berekenen