用 Python 编写一个程序,在给定的数据框中执行滚动窗口大小 3 的平均值计算

pythonpandasserver side programmingprogramming

假设,您有一个数据框,滚动窗口大小 3 的计算结果是,

滚动窗口的平均值是:
   Id Age  Mark
0 NaN NaN  NaN
1 1.5 12.0 85.0
2 2.5 13.0 80.0
3 3.5 13.5 82.5
4 4.5 31.5 90.0
5 5.5 60.0 87.5

为了解决这个问题,我们将遵循以下方法 −

解决方案

  • 定义一个数据框

  • 应用 df.rolling(window=2).mean() 计算滚动窗口大小 3 的平均值

df.rolling(window=2).mean()

示例

让我们检查以下代码以获得更好的理解−

import pandas as pd
df = pd.DataFrame({"Id":[1, 2, 3, 4, 5,6],
                     "Age":[12, 12, 14, 13, 50,70],
                     "Mark":[80, 90, 70, 95, 85,90],
                  })
print("Dataframe 为:\n",df)
print("滚动窗口的平均值是:\n",df.rolling(window=2).mean())

输出

Dataframe 为:
 Id Age Mark
0 1 12  80
1 2 12  90
2 3 14  70
3 4 13  95
4 5 50  85
5 6 70  90
滚动窗口的平均值是:
   Id Age Mark
0 NaN  NaN  NaN
1 1.5 12.0 85.0
2 2.5 13.0 80.0
3 3.5 13.5 82.5
4 4.5 31.5 90.0
5 5.5 60.0 87.5

相关文章