Новый столбец с статистикой по группам (функция трансформ), Анализ данных Python