Panda's: een staafdiagram maken vanuit groupby

U kunt de volgende syntaxis gebruiken om een staafdiagram te maken van een GroupBy-functie in panda’s:

 #calculate sum of values by group
df_groups = df. groupby ([' group_var '])[' values_var ']. sum ()

#create bar plot by group
df_groups. plot (kind=' bar ')

Het volgende voorbeeld laat zien hoe u deze syntaxis in de praktijk kunt gebruiken.

Voorbeeld: maak een staafdiagram van GroupBy in Pandas

Stel dat we het volgende panda’s DataFrame hebben dat de punten weergeeft die zijn gescoord door basketbalspelers van verschillende teams:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'A',
                            'B', 'B', 'B', 'B', 'B',
                            'C', 'C', 'C', 'C', 'C'],
                   ' points ': [12, 29, 34, 14, 10, 11, 7, 36,
                              34, 22, 41, 40, 45, 36, 38]})

#view first five rows of DataFrame
df. head ()

	team points
0 to 12
1 to 29
2 to 34
3 to 14
4 to 10

We kunnen de volgende syntaxis gebruiken om de som van de door elk team gescoorde punten te berekenen en een staafdiagram te maken om de som van elk team te visualiseren:

 import matplotlib. pyplot as plt

#calculate sum of points for each team
df. groupby (' team ')[' points ']. sum ()

#create bar plot by group
df_groups. plot (kind=' bar ')

Op de x-as staat de naam van elk team en op de y-as staat de som van de door elk team gescoorde punten.

We kunnen ook de volgende code gebruiken om de plot een beetje beter te maken:

 import matplotlib. pyplot as plt

#calculate sum of points for each team
df_groups = df. groupby ([' team '])[' points ']. sum ()

#create bar plot with custom aesthetics
df_groups. plot (kind=' bar ', title=' Total Points by Team ',
               ylabel=' Total Points ' , xlabel=' Team ', figsize=( 10,6 ) )

#rotate x-axis ticks vertically
plt. xticks (rotation= 0 )

Let op : U kunt hier de volledige documentatie voor de GroupBy- functie vinden.

Aanvullende bronnen

In de volgende tutorials wordt uitgelegd hoe u andere veelvoorkomende bewerkingen in panda’s kunt uitvoeren:

Panda’s: unieke waarden per groep tellen
Panda’s: hoe u de modus per groep kunt berekenen
Panda’s: hoe de correlatie per groep te berekenen