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

Python Pandas - 합계를 기준으로 DataFrame의 열 필터링

<시간/>

열의 합계를 기준으로 필터링하려면 loc()을 사용합니다. 방법. 여기 우리의 예에서 각 학생의 점수를 합산하여 400점 이상, 즉 80%인 학생 열을 얻습니다.

먼저 학생 기록으로 DataFrame을 만듭니다. 3명의 학생의 성적 기록이 있습니다. 즉 3개의 열 -

dataFrame =pd.DataFrame({ 'Jacob_Marks':[95, 90, 75, 85, 88],'Ted_Marks':[60, 50, 65, 85, 70],'Jamie_Marks':[77, 76 , 65, 45, 50]})

열을 기준으로 필터링합니다. 총점 400점 이상인 학생 가져오기 -

dataFrame =dataFrame.loc[:, dataFrame.sum(축=0)> 400]

예시

다음은 전체 코드입니다 -

pandas를 pd로 가져오기# 3개의 열이 있는 데이터 프레임 생성dataFrame =pd.DataFrame({ 'Jacob_Marks':[95, 90, 75, 85, 88],'Ted_Marks':[60, 50, 65, 85, 70 ],'Jamie_Marks':[77, 76, 65, 45, 50]})print"Dataframe...\n",dataFrame# 열을 기반으로 필터링# 총 점수가 400 이상인 학생 가져오기dataFrame =dataFrame.loc[ :, dataFrame.sum(axis=0)> 400]# dataframeprint"업데이트된 데이터 프레임...\n",dataFrame

출력

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

데이터 프레임... Jacob_Marks Jamie_Marks Ted_Marks0 95 77 601 90 76 502 75 65 653 85 45 854 88 50 70데이터 프레임 업데이트... Jacob_Marks0 951 902 758 854