找到Python Pandas中的滚动平均数

要找到滚动平均，我们将在Pandas中使用apply()函数。首先，让我们导入所需的库−

import pandas as pd

创建包含2列的DataFrame。其中一列是int列−

dataFrame = pd.DataFrame(
    {
        "Car": ['Tesla', 'Mercedes', 'Tesla', 'Mustang', 'Mercedes', 'Mustang'],
        "Reg_Price": [5000，1500，6500，8000，9000，6000]

    }
)

使用GroupBy进行分组，并使用apply()找到滚动平均值−

dataFrame.groupby("Car")["Reg_Price"].apply(
lambda x: x.rolling(center=False, window=2).mean())

例子

以下是代码−

import pandas as pd

# 创建DataFrame
dataFrame = pd.DataFrame(
   {
      "Car": ['Tesla', 'Mercedes', 'Tesla', 'Mustang', 'Mercedes', 'Mustang'],
      "Reg_Price": [5000, 1500, 6500, 8000, 9000, 6000]
   }
)

print"DataFrame ...\n",dataFrame

print"\n滚动平均数...\n",dataFrame.groupby("Car")["Reg_Price"].apply(
lambda x: x.rolling(center=False, window=2).mean())

输出

这将产生以下输出−

DataFrame ...
        Car    Reg_Price
0     Tesla        5000
1  Mercedes        1500
2     Tesla        6500
3   Mustang        8000
4  Mercedes        9000
5   Mustang        6000

滚动平均数...
0       NaN
1       NaN
2    5750.0
3       NaN
4    5250.0
5    7000.0
Name: Reg_Price, dtype: float64