Оставить группы по условию

Question

Нужно оставить все заказы только тех клиентов, у которых не меньше трёх заказов. Какой вариант верный?

Техпрогон · Accepted Answer

Правильный ответ: df.groupby("user").filter(lambda g: len(g) = 3). filter оставляет строки групп, для которых функция вернула True, исходные строки сохраняются. df.groupby("user").filter(lambda g: len(g) = 3). filter оставляет строки групп, для которых функция вернула True, исходные строки сохраняются. Вариант с size() = 3 вернёт булеву Series по клиентам, а не заказы. Маска из count() имеет индекс клиентов и не выровняется со строками. head(3) просто обрежет каждую группу до трёх строк. Быстрая альтернатива filter: df[df.groupby("user")["user"].transform("size") = 3].

Оставить группы по условию

Вопрос

Попробуйте сами

Коротко

Варианты ответа

Разбор

Термины по теме

Ещё по теме «groupby и агрегаты»