Найти медиану группы данных и обратиться к членам этой группы

Question

Как получить медиану группы, а затем применить ее ко всем членам группы?

У меня есть большой фрейм данных, для которого в столбце 1 указана группа, к которой они принадлежат.столбец 2 и их оценки в столбце 3.

Я хотел бы получить медиану группы, а затем назначить ее в столбце 4 членам этой группы.Разные медианы для каждой группы.

Answer 1

Используйте to_numeric с errors='coerce' для преобразования строк в числовое представление, а также для преобразования не числовых значений в NaN с, затем используйте GroupBy.transform с median дляНовый столбец в исходных данных заполнен агрегатными значениями:

df['score'] = pd.to_numeric(df['score'], errors='coerce')
df['median'] = df.groupby('col2')['score'].transform('median')

Answer 2

Вы можете использовать преобразование для добавления агрегата.

df['column4'] = df.groupby('column2')['column3'].transform('median')