pandas.core.groupby.SeriesGroupBy.filter

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

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

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

Параметры:

func:функция

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

dropna:bool

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

Возвращает:

Series

Примечания

Функции, изменяющие переданный объект, могут привести к неожиданному поведению или ошибкам и не поддерживаются. Подробнее см. Изменение с помощью пользовательских функций (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')
>>> df.groupby('A').B.filter(lambda x: x.mean() > 3.)
1    2
3    4
5    6
Name: B, dtype: int64