Pandas GroupBy.apply Дублирует первую группу: понимание поведения

При использовании функции groupby в Pandas может показаться, что метод apply применяет функцию дважды к первой строка фрейма данных. Такое поведение, хотя и кажется неожиданным, является задуманным.

Цель двойного приложения

Функция apply должна определить форму данных, которые она будет объединять. Для этого он дважды вызывает назначенную функцию — в данном случае checkit. Первый вызов помогает определить форму выходных данных, а второй выполняет операцию над группой.

Как избежать двойного эффекта

В зависимости от вашего варианта использования вы можете избежать двойного применения, используя альтернативу функции:

• агрегат: Требуется, чтобы возвращаемое значение было сводной статистикой, например средним или суммой.
• transform: Требует, чтобы возвращаемое значение имело ту же форму, что и входная группа.
• filter: Требует, чтобы возвращаемое значение было логическим индексом, указывающим, какие строки следует сохранить.

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

Соображения о побочных эффектах

Если применяемая функция не имеет побочных эффектов, т. е. он не изменяет исходный фрейм данных — тогда двойное приложение, скорее всего, не имеет значения. Однако если функция выполняет какие-либо манипуляции с данными, двойное применение к первой строке может привести к непредвиденным последствиям.