Панды - скользящая средняя, сгруппированная по нескольким столбцам
Новичок в Пандах, так что терпите меня.
Мой фрейм данных имеет формат
date,name,country,tag,cat,score
2017-05-21,X,US,free,4,0.0573
2017-05-22,X,US,free,4,0.0626
2017-05-23,X,US,free,4,0.0584
2017-05-24,X,US,free,4,0.0563
2017-05-21,X,MX,free,4,0.0537
2017-05-22,X,MX,free,4,0.0640
2017-05-23,X,MX,free,4,0.0648
2017-05-24,X,MX,free,4,0.0668
Я пытаюсь найти способ найти скользящее среднее значение X дня в группе стран / тегов / категорий, поэтому мне нужно:
date,name,country,tag,cat,score,moving_average
2017-05-21,X,US,free,4,0.0573,0
2017-05-22,X,US,free,4,0.0626,0.0605
2017-05-23,X,US,free,4,0.0584,0.0594
2017-05-24,X,US,free,4,0.0563,and so on
...
2017-05-21,X,MX,free,4,0.0537,and so on
2017-05-22,X,MX,free,4,0.0640,and so on
2017-05-23,X,MX,free,4,0.0648,and so on
2017-05-24,X,MX,free,4,0.0668,and so on
Я пробовал что-то в строках группировки по нужным мне столбцам, а затем использовал pd.rolling_mean, но в итоге получилось несколько NaN
df.groupby(['date', 'name', 'country', 'tag'])['score'].apply(pd.rolling_mean, 2, min_periods=2) # window size 2
Как бы я поступил правильно?
1 ответ
Решение
IIUC:
(df.assign(moving_score=df.groupby(['name','country','tag'], as_index=False)[['score']]
.rolling(2, min_periods=2).mean().fillna(0)
.reset_index(0, drop=True)))
Выход:
date name country tag cat score moving_score
0 2017-05-21 X US free 4 0.0573 0.00000
1 2017-05-22 X US free 4 0.0626 0.05995
2 2017-05-23 X US free 4 0.0584 0.06050
3 2017-05-24 X US free 4 0.0563 0.05735
4 2017-05-21 X MX free 4 0.0537 0.00000
5 2017-05-22 X MX free 4 0.0640 0.05885
6 2017-05-23 X MX free 4 0.0648 0.06440
7 2017-05-24 X MX free 4 0.0668 0.06580