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

Python Pandas - 합계를 기준으로 DataFrame에서 몇 개의 행 필터링

<시간/>

합계를 기준으로 DataFrame에서 몇 개의 행을 필터링하기 위해 Student Marks의 예를 고려했습니다. 합계가 200보다 큰 특정 과목의 합계를 계산해야 합니다. 즉, 해당 특정 과목의 세 학생의 합계가 200보다 큰 경우입니다. 이러한 방식으로 총계가 200보다 작은 행을 맞출 수 있습니다.

먼저 3개의 열, 즉 3명의 학생에 대한 레코드가 있는 DataFrame을 만듭니다.

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

행을 기준으로 필터링합니다. 3명의 학생 모두에 대해 합계가 200보다 큰 행을 가져오는 중 -

dataFrame =dataFrame[dataFrame.sum(axis=1)> 200]

예시

다음은 전체 코드입니다 -

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

출력

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

Dataframe... Jacob_Marks Jamie_Marks Ted_Marks0 95 77 601 90 76 502 70 60 653 85 45 854 88 50 5 4 5 4 5 4 5 4 5 4 5 4 5 4 업데이트된 데이터프레임... Jacob_Marks Jamie_Marks Ted_96 06 05 77