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

주어진 데이터 프레임에서 롤링 창 크기 3의 평균 계산을 수행하는 프로그램을 Python으로 작성하십시오.

<시간/>

데이터 프레임이 있고 롤링 창 크기 3 계산의 결과는 다음과 같다고 가정합니다.

Average of rolling window is:
   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
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 is:\n",df)
print("Average of rolling window is:\n",df.rolling(window=2).mean())

출력

Dataframe is:
 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
Average of rolling window is:
   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