• Exemplo 1: Obter grupo após usar groupby()
• Exemplo 2: Obter colunas específicas do grupo depois de usar groupby()

É possível utilizar os seguintes métodos para obter um grupo específico depois de utilizar o groupby() num DataFrame do pandas:

Método 1: Obter grupo após utilizar groupby()

 grouped_df. get_group (' A ') 

Método 2: Obter colunas específicas do grupo depois de utilizar groupby()

 grouped_df[[' column1 ', ' column3 ']]. get_group (' A ') 

Os exemplos seguintes mostram como utilizar cada método na prática com o seguinte DataFrame do pandas:

 import pandas as pd #create DataFrame df = pd. DataFrame ({' store ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B', 'B'], ' sales ': [12, 15, 24, 24, 14, 19, 12, 38], ' refunds ': [4, 8, 7, 7, 10, 5, 4, 11]}) #view DataFrame print (df) store sales refunds 0 A 12 4 1 A 15 8 2 A 24 7 3 A 24 7 4 B 14 10 5 B 19 5 6 B 12 4 7 B 38 11 

Exemplo 1: Obter grupo após usar groupby()

O código a seguir mostra como usar o groupby( ) para agrupar as linhas por nome de loja e, em seguida, usar a função get_group() para obter todas as linhas que pertencem ao grupo com o nome de grupo "A":

 #agrupar linhas de DataFrame com base no valor da coluna 'store' grouped_stores = df. groupby ([' store ']) #obter todas as linhas que pertencem ao nome do grupo 'A' grouped_stores. get_group (' A ') store sales refunds 0 A 12 4 1 A 15 8 2 A 24 7 3 A 24 7 

Observe que get_group() devolve todas as linhas que pertencem ao grupo com o nome de grupo "A".

Exemplo 2: Obter colunas específicas do grupo depois de usar groupby()

O código a seguir mostra como usar o groupby( ) para agrupar as linhas por nome de loja e, em seguida, usar a função get_group() para obter todas as linhas que pertencem ao grupo com o nome de grupo "A" apenas para as colunas "vendas" e "reembolsos":

 #agrupar linhas de DataFrame com base no valor da coluna 'store' grouped_stores = df. groupby ([' store ']) #obter todas as linhas que pertencem ao nome do grupo 'A' para as colunas de vendas e reembolsos grouped_stores[[[' store ', ' refunds ']]. get_group (' A ') store refunds 0 A 4 1 A 8 2 A 7 3 A 7 

Observe que get_group() devolve todas as linhas que pertencem ao grupo com o nome de grupo "A" apenas para as colunas "vendas" e "reembolsos".