用 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