[pandas-basic] 조건 검색

판다스에서 조건으로 행을 선택하는 방법을 알아보겠습니다.

import pandas as pd
df = pd.read_excel('census.xlsx')

 

쿼리

query 메소드에 코드를 문자열 형태로 입력하면, 그 코드가 참인 행을 선택해줍니다. 코드에서 열 이름은 변수처럼 사용하면 됩니다. 아래는 age 열의 값이 40보다 큰 행만 선택합니다.

df.query('age > 40')

 

and

두 가지 조건이 동시에 참이어야 하는 경우에는 and를 사용합니다. 아래는 age가 40이 넘고 sex는 Male인 경우를 찾습니다.

df.query('age > 40 and sex == "Male"')

 

or

둘 중에 하나라도 참이어야 하는 경우에는 or를 사용합니다. 아래는 relationship이 Husband이거나 또는 Wife인 경우를 찾습니다.

df.query('relationship == "Husband" or relationship == "Wife"')

 

not

반대되는 경우를 찾으려면 not을 사용합니다. 아래는 age가 30보다 작고 race가 Black이 경우가 아닌 행를 찾습니다.

df.query('not (age < 30 and race == "Black")')

 

in

파이썬에서 in 연산자는 포함 관계를 나타냅니다. 아래는 relationship의 값이 Husband 또는 Wife에 포함되는 경우를 뜻합니다. or와 비슷하지만 비교할 대상이 많을 경우에는 포함관계의 형태로 나타내면 코드가 간결합니다.

df.query('relationship in ["Husband", "Wife"]')

반대로 포함되지 않음을 나타낼 때는 not in을 씁니다.

df.query('relationship not in ["Husband", "Wife"]')