Что делает функция transform при использовании с groupby в Pandas

Функция transform после groupby применяет функцию к каждой группе и возвращает объект с таким же размером, как исходный DataFrame. Результат выравнивается по индексу исходных данных, что удобно для создания новых столбцов на основе групповых агрегаций.

Пример:

import pandas as pd

df = pd.DataFrame({'group': ['A', 'A', 'B', 'B'], 'value': [1, 2, 3, 4]})
df['group_mean'] = df.groupby('group')['value'].transform('mean')

Результат:

  group  value  group_mean
0     A      1         1.5
1     A      2         1.5
2     B      3         3.5
3     B      4         3.5

Ключевые особенности:

• Сохраняет исходную форму данных
• Автоматически выравнивает результат по индексу
• Полезен для нормализации внутри групп или создания признаков
• Работает с агрегирующими функциями (mean, sum) и пользовательскими функциями