gpt4 book ai didi

python - 在 Python 中计算滚动总和

转载 作者:行者123 更新时间:2023-12-01 09:09:00 24 4
gpt4 key购买 nike

我有一个具有以下格式的数据框 df:

    Type     2013         2014          2015         2016          2017         
EU NO_EU EU NO_EU EU NO_EU EU NO_EU EU NO_EU
A 4 6 2 1 2 7 1 4 2 6

B 3 0 8 3 11 22 2 2 2 1

我想创建一个新的数据框,其中包含 EU 和 NO_EU 3 年期间的滚动总和。因此,我的新数据框应该是这样的:

           Type     2013         2014          2015         2016          2017         
EU NO_EU EU NO_EU EU NO_EU EU NO_EU EU NO_EU

A - - - - 8 14 5 12 5 17
B - - - - 22 25 21 27 15 25

前两年将没有数据,因为我们没有足够的数据来计算过去 3 年的总和。

关于如何在 python 中做到这一点有什么想法吗?也许使用类似于 df.rolling(window=3).sum 的东西?我的实际数据框有很多行和列,因此我将使用循环来计算它们。

感谢您提前提供的帮助

最佳答案

您可以尝试使用rollingaxis=1,即沿列,而不是沿行axis=0

df.rolling(window=3,axis=1).sum()

关于python - 在 Python 中计算滚动总和,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51820011/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com