본문 바로가기
Python/Python Examples

[Python Examples] pd.DataFrame(): Section별 학생 구분하여 출력하기

by Henry Cho 2022. 10. 28.
728x90

pd.DataFrame(): Section별 학생 구분하여 출력하기

포스트 난이도: HOO_Intern


# Example Codes

 

import pandas as pd
df = pd.DataFrame({"section": [3,1,1,2,2,3], "students": ['James', 'Julia', 'Megan', 'Henry', 'Minji', 'Yelin']})
sec_1 = df[df['section'] == 1]
sec_2 = df[df['section'] == 2]
sec_3 = df[df['section'] == 3]

print(sec_1)
print(sec_2)
print(sec_3)

   section students
1        1    Julia
2        1    Megan
   section students
3        2    Henry
4        2    Minji
   section students
0        3    James
5        3    Yelin

Pandas의 pd.DataFrame()을 통해서 특정 인덱스에 맞는 데이터를 출력할 수 있다.

위의 예제 코드를 살펴보면 1부터 3까지 섹션으로 나눠져 있으며, 각 섹션마다 학생들이 구분되어 있는 걸 알 수 있다.

pd.DataFrame()에 {}을 작성해주고 {} 안에 쉼표로 각 인덱스를 구분하여 넣어주면 된다.

pd.DataFrame()에서 위의 코드처럼 섹션별 구분을 해준다면 작성 순서를 유의해야 한다.

 

예를 들어서 섹션 숫자가 3,1,1,2,2,3으로 작성되어 있기 때문에 첫 번째에 나와 있는 3 섹션은 제임스(James)가 해당된다.

마찬가지로 마지막에서 두번째에 위치한 섹션 2는 헨리(Henry)가 된다.

따라서 작성된 순서가 학생과 섹션 숫자를 연결해주고 있으며, 마치 배열 작성 방법을 생각하면 이해하기 수월하다.

 

예제 코드 하단에 출력된 결과를 살펴볼 수 있다.

출력된 결과에는 섹션 숫자와 해당 섹션에 맞는 학생들이 구분되어 출력된 걸 확인할 수 있다.

왼쪽에 출력된 숫자는 인덱스 또는 배열 순서를 의미하는 숫자이다.


 

728x90

댓글