Панды создают столбец для среднего количества сгруппированных

Question

У меня есть сгруппированный фрейм данных, который выглядит следующим образом:

 player_id  shot_type  count
    01         03        3
    02         01        3
               03        2
    03         01        4

Я хочу добавить дополнительный столбец, представляющий собой среднее значение счетчиков shot_type по player_id, который будет выглядеть следующим образом:

 player_id  shot_type  count  mean_shot_type_count_player
    01         03        3            (3+2)/2
    02         01        3            (3+4)/2
               03        2            (3+2)/2
    03         01        4            (3+4)/2

Answer 1

Использование GroupBy.transform :

df['mean_shot_type_count_player']=df.groupby('shot_type')['count'].transform('mean')
print(df)

Вывод:

   player_id  shot_type  count  mean_shot_type_count_player
0         01         03      3                          2.5
1         02         01      3                          3.5
2                    03      2                          2.5
3         03         01      4                          3.5