Вычислить изменение от базовой линии по идентификатору

Question

У меня есть некоторые продольные данные, которые представляют собой animal_id по дням обучения, как показано ниже:

Как мне создать столбец, который вычисляет изменение от базовой линии для каждого animal_id?Здесь базовый уровень будет где ord = 0?

Answer 1

Для этого, если вы не хотите сортировать данные, вы можете использовать суррогатный фрейм данных со значением веса тела для ord = 0, а затем объединить его с предыдущим фреймом данных.

df_ord = df.query("ord==0").rename(columns={'body_weight':'body_weight_base'})
df_ord = df_ord.drop('ord',axis=1)
df = df.merge(df_ord)
df['change'] = df['body_weight'] - df['body_weight_base']

Answer 2

Используя transform first, обратите внимание на это, если ваш df уже отсортирован

df['New']=df['Body_Weight']-df.groupby('Animal_id')['Body_weight'].transform('first')