[Pandas] 데이터프레임 순회하기

[Pandas] 데이터프레임 순회하기

특정 조건에 맞는 데이터 프레임 값을 출력해야하는 경우가 있다.

데이터프레임 순회를 하여 데이터프레임을 모두 출력하는 방법에 대해서 알아보자

 

먼저 샘플 데이터 프레임을 생성해준다.

import pandas as pd
import names
from random import *

age = [randint(16, 19) for i in range(0, 5)]
gender = [choice(["male", "female"]) for i in range(0, 5)]
name = [names.get_first_name(gender=g) for g in gender]
grade = [choice(['A','B','C','D']) for i in range(0, 5)]
math = [randint(50, 100) for i in range(0, 5)]
english = [randint(50, 100) for i in range(0, 5)]

dic = {"age":age, "gender":gender, "name":name, "class":grade, "math":math, "english":english}
df = pd.DataFrame(dic)

df

 

행(row) 방향으로 순회 

iterrows()는 각 데이터프레임의 행의 정보를 담은 객체라고 보면 된다.

iterrows()를 사용해서 for loop를 돌려 원하는 행의 정보를 가져온다.

for idx, row in df.iterrows():
    print(f"{idx} name:{row['name']} | gender:{row['gender']} | age:{row['age']}")

 

열(column) 방향으로 순회

df.columns에 데이터프레임 열 이름이 배열 형태로 존재하므로, 이를 이용하면 쉽게 순회가 가능하다.

for col in df.columns:
    dtype = str(df[col].dtype)
    if dtype == 'object':
        print(f"column:{col}, data:{df[col].mode().values[0]}")

 

참고

https://zephyrus1111.tistory.com/172