首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >Pandas -计算组的滚动平均值,不包括当前行

Pandas -计算组的滚动平均值,不包括当前行
EN

Stack Overflow用户
提问于 2019-02-11 07:09:27
回答 1查看 3.2K关注 0票数 3

举个例子:

代码语言:javascript
运行
复制
data = {'Platoon': ['A','A','A','A','A','A','B','B','B','B','B','C','C','C','C','C'],
        'Date' : [1,2,3,4,5,6,1,2,3,4,5,1,2,3,4,5],
       'Casualties': [1,4,5,7,5,5,6,1,4,5,6,7,4,6,4,6]}
df = pd.DataFrame(data)

这将计算滚动平均值,包括当前行:

代码语言:javascript
运行
复制
df['avg'] = df.groupby(['Platoon'])['Casualties'].transform(lambda x: x.rolling(2, 1).mean())

这就给出了:

代码语言:javascript
运行
复制
Platoon   Date Casualties Avg  
        A   1   1         1.0
        A   2   4         2.5
        A   3   5         4.5
        A   4   7         6.0
        ......

我想要得到的是:

代码语言:javascript
运行
复制
Platoon   Date Casualties Avg  
        A   1   1         1.0
        A   2   4         1.0
        A   3   5         2.5
        A   4   7         4.5
        ......

我想我可以在这里使用shift,但是我不能理解它!

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2019-02-11 07:30:03

您需要使用带有bfillshift

代码语言:javascript
运行
复制
df.groupby(['Platoon'])['Casualties'].apply(lambda x: x.rolling(2, 1).mean().shift().bfill())
票数 5
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/54622062

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档