[python-ds] 판다스: 조건 검색

 

참/거짓으로 선택

pd.options.display.max_rows = 6

df['age'] > 40

0        False
1         True
         ...
32559    False
32560     True
Name: age, Length: 32561, dtype: bool

df[df['age'] > 40]

 

쿼리

df.query('age > 40')

 

AND

AND는 &로 표시. 괄호에 유의.

40세 넘는 남자

(df['age'] > 40) & (df['sex'] == 'Male')

0        False
1         True
2        False
         ...
32558    False
32559    False
32560    False
Length: 32561, dtype: bool

df[(df['age'] > 40) & (df['sex'] == 'Male')]

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

 

OR

OR는 |로 표시. 괄호에 유의.

남편 또는 아내

(df['relationship'] == 'Husband') | (df['relationship'] == 'Wife')

0        False
1         True
2        False
         ...
32558    False
32559    False
32560     True
Name: relationship, Length: 32561, dtype: bool

df[(df['relationship'] == 'Husband') | (df['relationship'] == 'Wife')]

df['relationship'].unique()

array(['Not-in-family', 'Husband', 'Wife', 'Own-child', 'Unmarried',
       'Other-relative'], dtype=object)

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

 

isin

df['relationship'].isin(['Husband', 'Wife'])

0        False
1         True
2        False
         ...
32558    False
32559    False
32560     True
Name: relationship, Length: 32561, dtype: bool

df[df['relationship'].isin(['Husband', 'Wife'])]

5 in [1, 2, 5]

True

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

 

NOT

NOT은 ~로 표시

남편/아내 제외

~df['relationship'].isin(['Husband', 'Wife'])

0         True
1        False
2         True
         ...
32558     True
32559     True
32560    False
Name: relationship, Length: 32561, dtype: bool

df[~df['relationship'].isin(['Husband', 'Wife'])]

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

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

in의 경우에는 not in도 가능

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