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

DataFrame에서 'A' 학년 학생의 이름을 출력하는 프로그램을 Python으로 작성하십시오.

<시간/>

입력 -

Assume, you have DataFrame,
 Id  Name Grade
0 1 stud1   A
1 2 stud2   B
2 3 stud3   C
3 4 stud4   A
4 5 stud5   A

출력 -

그리고 'A' 학년 학생 이름에 대한 결과,

0    stud1
3    stud4
4    stud5

해결책

이를 해결하기 위해 다음과 같은 접근 방식을 따릅니다.

  • DataFrame 정의

  • 값을 DataFrame과 비교

df[df['Grade']=='A']
  • 결과를 다른 DataFrame에 저장하고 Name을 가져옵니다.

예시

더 나은 이해를 위해 다음 구현을 살펴보겠습니다.

import pandas as pd
data =
[[1,'stud1','A'],[2,'stud2','B'],[3,'stud3','C'],[4,'stud4','A'],[5,'stud5','A']]
df = pd.DataFrame(data,columns=('Id','Name','Grade'))
print("DataFrame is\n",df)
print("find the A grade students name\n")
result = df[df['Grade']=='A']
print(result['Name'])

출력

DataFrame is
 Id Name Grade
0 1 stud1  A
1 2 stud2  B
2 3 stud3  C
3 4 stud4  A
4 5 stud5  A
find the A grade students name
0    stud1
3    stud4
4    stud5
Name: Name, dtype: object