groupby: agg или transform

Question

Нужно добавить к каждой строке заказов колонку с суммой заказов её клиента, не меняя число строк. Что выбрать?

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

Правильный ответ: df.groupby("user")["amount"].transform("sum"). transform возвращает результат той же длины и с тем же индексом, что исходный DataFrame, поэтому его можно сразу присвоить в колонку. agg("sum") и .sum() дают одну строку на клиента: присвоение в колонку выровняет по индексу и даст NaN или… df.groupby("user")["amount"].transform("sum"). transform возвращает результат той же длины и с тем же индексом, что исходный DataFrame, поэтому его можно сразу присвоить в колонку. agg("sum") и .sum() дают одну строку на клиента: присвоение в колонку выровняет по индексу и даст NaN или неверные значения. filter оставляет или выкидывает группы целиком, он не считает суммы.

groupby: agg или transform

Вопрос

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

Коротко

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

Разбор

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

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