Pandas Dataframe 조건문 쓰는 방법

DataFrame 조건문을 쓰는 방법에 대해 공유하고자 합니다.

 

특정 column(열) 에서 특정 value(값)을 가진 row(행)을 가져오고자 할 때 많이 쓰는 방법입니다.

 

예시로 dataframe을 하나 만들어서 여러 조건문을 붙여보겠습니다.

import pandas as pd

df1 = pd.DataFrame(columns=['id', 'values', 'condition'], 
                   data=[
                       ['a', 123, 'wow'],
                       ['b', 234, 'nooo'], 
                       ['c', 345, 'wow'],
                       ['d', 567, 'good']
                       ]
                    )

 

df1의 값은 이렇게 됩니다.

 

여기서 values의 값이 123인 행을 가져와 보겠습니다.

# "values" column 중 123 값을 가진 row를 가져온다
condition_df = df1[df1['values'] == 123]

 

이번엔 or 조건문으로 row를 가져와 보겠습니다.

# or 조건문
# "values" column 중 123 값을 가지거나(or) "condition" column 중 'nooo' 값을 가진 row를 가져온다
condition_df = df1[(df1['values'] == 123) | (df1['condition'] == 'nooo')]

 

and 조건문을 써서 row를 가져와보겠습니다!

# and 조건문
# "values" column 중 234 값을 가지면서(and) "condition" column 중 'nooo' 값을 가진 row를 가져온다
condition_df = df1[(df1['values'] == 234) & (df1['condition'] == 'nooo')]

 

 

 

도움이 되셨다면 아무 광고나 클릭 한 번 부탁드립니다👍