[Python] Boolean Indexing

특정 값이 최대값을 가지는 행 추출

---

In [2]:

import pandas as pd
import numpy as np
import seaborn as sns

df = sns.load_dataset('titanic')
df.head()

Out[2]:

boolean indexing

---

시리즈 객체에 특정 조건식을 적용해 해당 조건에 참인 행(row)을 추출하기 위해선 boolean indexing을 이용할 수 있다.

다음의 예제로 성별이 남성이면서 생존한 인원에 대한 행들을 추출해보자

In [19]:

condition = (df.sex == 'male') & (df.survived ==1) # 조건식 작성

df[condition]

Out[19]:

In [20]:

df.loc[condition] # loc를 이용해도 무방하다.

Out[20]:

이번에는 or 연산자를 이용해 pclass가 1 혹은 2인 행들을 추출해보자.

In [28]:

condition = (df.pclass == 1) | (df.pclass == 2)

df.loc[condition, ['survived', 'sex', 'pclass']] # 필터링 된 row의 특정 컬럼만 추출

Out[28]:

isin()

---

sibsp가 1, 2, 3의 값을 갖는 데이터만 추출하기 위해선 아래의 방법처럼 boolean indexing을 사용할 수 있다.

In [31]:

con1 = df['sibsp'] == 1
con2 = df['sibsp'] == 2
con3 = df['sibsp'] == 3

filtered_df = df.loc[con1 | con2 | con3, :]
filtered_df

Out[31]:

위처럼 특정한 리스트 형태의 값들을 포함하는 행을 추출하고 싶다면 아래와 같이 isin 함수를 사용하는 것이 더욱 간편하다.

In [32]:

isin_con = df['sibsp'].isin([1, 2, 3])
df.loc[isin_con]

Out[32]: