Pandy: jak zdobyć grupę po użyciu groupby()

Możesz użyć następujących metod, aby uzyskać określoną grupę po użyciu funkcji groupby() w ramce DataFrame pandy:

Metoda 1: Uzyskaj grupę po użyciu groupby()

 grouped_df. get_group (' A ')

Metoda 2: Uzyskaj określone kolumny z grupy po użyciu groupby()

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

Poniższe przykłady pokazują, jak w praktyce używać każdej metody z następującą ramką DataFrame pand:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' store ': ['A', 'A', 'A', 'A', '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 to 12 4
1 to 15 8
2 to 24 7
3 to 24 7
4 B 14 10
5 B 19 5
6 B 12 4
7 B 38 11

Przykład 1: Uzyskaj grupę po użyciu groupby()

Poniższy kod pokazuje, jak użyć funkcji groupby( ) do grupowania wierszy według nazwy sklepu, a następnie użyć funkcji get_group() do pobrania wszystkich wierszy należących do grupy o nazwie „A”:

 #group rows of DataFrame based on value in 'store' column
grouped_stores = df. groupby ([' store '])

#get all rows that belong to group name 'A'
grouped_stores. get_group (' A ')

    store sales refunds
0 to 12 4
1 to 15 8
2 to 24 7
3 to 24 7

Zauważ, że get_group() zwraca wszystkie wiersze należące do grupy o nazwie „A”.

Przykład 2: Pobierz określone kolumny z grupy po użyciu groupby()

Poniższy kod pokazuje, jak użyć funkcji groupby( ) do grupowania wierszy według nazwy sklepu, a następnie użyć funkcji get_group() do pobrania wszystkich wierszy należących do grupy o nazwie grupy „A” tylko dla kolumn „sprzedaż” i „Zwroty środków” :

 #group rows of DataFrame based on value in 'store' column
grouped_stores = df. groupby ([' store '])

#get all rows that belong to group name 'A' for sales and refunds columns
grouped_stores[[' store ', ' refunds ']]. get_group (' A ')

    store refunds
0 to 4
1 to 8
2 to 7
3 to 7

Należy pamiętać, że funkcja get_group() zwraca wszystkie wiersze należące do grupy o nazwie „A” tylko dla kolumn „sprzedaż” i „zwroty środków”.

Dodatkowe zasoby

Poniższe samouczki wyjaśniają, jak wykonywać inne typowe operacje na pandach:

Jak wykonać sumę GroupBy w Pandach
Jak korzystać z Groupby i Plot w Pandach
Jak liczyć unikalne wartości za pomocą GroupBy w Pandach