pandas.core.groupby.DataFrameGroupBy.filter

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

Возвращает копию DataFrame, исключая отфильтрованные элементы.

Элементы из групп отфильтровываются, если они не удовлетворяют булевому критерию, заданному func.

Параметры

func:функция

Функция, применяемая к каждой подрамке. Должна возвращать True или False.

dropna:Удалять группы, которые не прошли фильтрацию. По умолчанию True;

Если False, группы, оцениваемые как False, заполняются значениями NaN.

Возвращает

filtered:DataFrame

Примечания

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

Функции, которые изменяют переданный объект, могут привести к непредсказуемому поведению или ошибкам и не поддерживаются. См. Изменение с помощью пользовательских функций (UDF) методов для получения дополнительной информации.

Примеры

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