pandas.core.groupby.DataFrameGroupBy.filter

DataFrameGroupBy.filter(func, dropna=True, *args, **kwargs) [source]

Возвращает копию DataFrame, исключая элементы из групп, которые не удовлетворяют булевому критерию, заданному функцией.

Параметры:	f : функция Функция, применяемая к каждому подкадру. Должна возвращать True или False. dropna : Удалить группы, которые не прошли фильтр. По умолчанию True; если False, группы, которые вернули False, заполняются NaN.
Возвращает:	filtered : DataFrame

Примечания

Каждый подкадр получает атрибут ‘name’, в случае если вам нужно узнать, с какой группой вы работаете.

Примеры

>>> import pandas as pd
>>> df = pd.DataFrame({'A' : ['foo', 'bar', 'foo', 'bar',
...                           'foo', 'bar'],
...                    'B' : [1, 2, 3, 4, 5, 6],
...                    'C' : [2.0, 5., 8., 1., 2., 9.]})
>>> grouped = df.groupby('A')
>>> grouped.filter(lambda x: x['B'].mean() > 3.)
     A  B    C
1  bar  2  5.0
3  bar  4  1.0
5  bar  6  9.0