Python 求一个月内最大的 n 的平均值，但天必须是唯一的(Pandas )

发布于07月01日

我怎样才能找到一个月内n个最大值的平均值，但天必须是唯一的？

我确实也有一个时间戳列，但我猜制作它们的列是一种方式？

我试了df['peak_avg'] = df.groupby(['month', 'day'])['value'].transform(lambda x: x.nlargest(3).mean())天，但这是最大的三天的平均时间.

month	day	value	peak_avg (expected)
1	1	35	35
1	1	30	35
2	1	34	28.5
2	2	23	28.5
3	1	98	97
3	2	96.	97

推荐答案

IIUC，你可以在month和day列中删除重复项，最后填充它们

df['peak_avg'] = (df.sort_values(['month', 'day', 'value'], ascending=[True, True, False])
                  .drop_duplicates(['month', 'day'])
                  .groupby(['month'])['value']
                  .transform(lambda x: x.head(3).mean()))
df['peak_avg'] = df.groupby(['month', 'day'])['peak_avg'].apply(lambda g: g.ffill().bfill())

print(df)

   month  day  value  peak_avg
0      1    1     35      35.0
1      1    1     12      35.0
2      2    1     34      28.5
3      2    3     23      28.5
4      3    1     98      98.0
5      3    2     98      98.0