Метод get_group
Метод get_group применяется к объекту DataFrameGroupBy,
который возвращается после группировки с помощью groupby.
Он извлекает одну конкретную группу по её уникальному значению.
Первым и единственным параметром передаётся ключ группы,
которую нужно получить.
Синтаксис
grouped.get_group(name)
Пример
Давайте сгруппируем таблицу по столбцу 'b' и выберем группу со значением 20:
import pandas as pd
df = pd.DataFrame({
'a': [1, 2, 3, 4, 5],
'b': [10, 20, 10, 20, 10]
})
grouped = df.groupby('b')
res = grouped.get_group(20)
print(res)
Результат выполнения кода:
a b
1 2 20
3 4 20
Пример
Если группа была создана по нескольким столбцам, в метод передаётся кортеж с соответствующими значениями:
import pandas as pd
df = pd.DataFrame({
'a': [1, 2, 3, 4, 5],
'b': [10, 20, 10, 20, 10],
'c': ['x', 'y', 'x', 'y', 'z']
})
grouped = df.groupby(['b', 'c'])
res = grouped.get_group((10, 'x'))
print(res)
Результат выполнения кода:
a b c
0 1 10 x
2 3 10 x
Пример
Попытка получить несуществующую группу вызовет ошибку. Поэтому перед вызовом можно проверить наличие ключа:
import pandas as pd
df = pd.DataFrame({
'a': [1, 2, 3, 4, 5],
'b': [10, 20, 10, 20, 10]
})
grouped = df.groupby('b')
key = 30
if key in grouped.groups:
res = grouped.get_group(key)
print(res)
else:
print("Group not found")
Результат выполнения кода:
"Group not found"
