pandas.core.groupby.DataFrameGroupBy.filter

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

Фильтр элементов из групп, которые не удовлетворяют критерию.

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

Параметры:

func:функция

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

dropna:bool

Удалить группы, которые не прошли фильтр. По умолчанию True; если False, группы, которые выдали False, заполняются значениями NaN.

Возвращает:

DataFrame

Примечания

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

Функции, изменяющие переданный объект, могут привести к неожиданному поведению или ошибкам и не поддерживаются. Подробнее см. Изменение с помощью пользовательских функций (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