Pandas: como criar um gráfico de barras a partir de groupby

Você pode usar a seguinte sintaxe para criar um gráfico de barras a partir de uma função GroupBy no pandas:

 #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 ')

O exemplo a seguir mostra como usar essa sintaxe na prática.

Exemplo: Crie um gráfico de barras de GroupBy no Pandas

Suponha que temos o seguinte DataFrame do pandas que mostra os pontos marcados por jogadores de basquete de diferentes times:

 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

Podemos usar a seguinte sintaxe para calcular a soma dos pontos marcados por cada equipe e criar um gráfico de barras para visualizar a soma de cada equipe:

 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 ')

O eixo x mostra o nome de cada equipe e o eixo y mostra a soma dos pontos marcados por cada equipe.

Também podemos usar o seguinte código para tornar o gráfico um pouco melhor:

 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 )

Nota : Você pode encontrar a documentação completa da função GroupBy aqui .

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras operações comuns em pandas:

Pandas: como contar valores únicos por grupo
Pandas: como calcular o modo por grupo
Pandas: como calcular a correlação por grupo