Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

롤링 평균 찾기 – Python Pandas

<시간/>

롤링 평균을 찾기 위해 Pandas에서 apply() 함수를 사용할 것입니다. 먼저 필요한 라이브러리를 가져오도록 합시다 -

pandas를 pd로 가져오기

2개의 열이 있는 DataFrame을 만듭니다. 하나는 int 열입니다 -

dataFrame =pd.DataFrame( { "자동차":['테슬라', '메르세데스', '테슬라', '머스탱', '메르세데스', '머스탱'], "Reg_Price":[5000, 1500, 6500 , 8000, 9000, 6000] })

GroupBy를 사용하여 그룹화하고 apply() −

를 사용하여 롤링 평균 찾기
dataFrame.groupby("자동차")["Reg_Price"].apply(lambda x:x.rolling(center=False, window=2).mean())

예시

다음은 코드입니다 -

pandas를 pd로 가져오기# Create DataFramedataFrame =pd.DataFrame( { "Car":['Tesla', 'Mercedes', 'Tesla', 'Mustang', 'Mercedes', 'Mustang'], "Reg_Price":[5000, 1500, 6500, 8000, 9000, 6000] })print"DataFrame ...\n",dataFrameprint"\n롤링 평균...\n",dataFrame.groupby("자동차")["Reg_Price"] .apply(람다 x:x.rolling(center=False, window=2).mean())

출력

이것은 다음과 같은 출력을 생성합니다 -

DataFrame ... 자동차 Reg_Price0 Tesla 50001 Mercedes 15002 Tesla 65003 Mustang 80004 Mercedes 90005 Mustang 6000Rolling Mean...0 NaN1 NaN2 5750.03 NaN4 5250.05 ,Regtype:float6, 7000.0Name