查詢滾動平均 – Python Pandas
要查詢滾動均值,我們將使用 Pandas 中的 apply() 函式。首先,讓我們匯入所需的庫 −
import pandas as pd
建立一個帶有 2 列的資料框。其中一列是一個 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 # Create DataFrame dataFrame = pd.DataFrame( { "Car": ['Tesla', 'Mercedes', 'Tesla', 'Mustang', 'Mercedes', 'Mustang'], "Reg_Price": [5000, 1500, 6500, 8000, 9000, 6000] } ) print"DataFrame ...\n",dataFrame print"\nRolling Mean...\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 Rolling Mean... 0 NaN 1 NaN 2 5750.0 3 NaN 4 5250.0 5 7000.0 Name: Reg_Price, dtype: float64
廣告