pandas.core.groupby.DataFrameGroupBy.filter

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

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

Параметры:	f : function Функция, применяемая к каждому подфрейму. Должна возвращать True или False. dropna : Drop groups that do not pass the filter. True by default; Если False, группы, которые дают результат False, заполняются NaN.
Возвращает:	filtered : DataFrame

Примечания

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

Примеры

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