(python) 칼럼의 값(2개 이상)에 따라 행 뽑기
OR 과 isin()을 이용한 인덱싱
(python) 칼럼의 값에 따라 행 뽑기 (열과 행 혼합 인덱싱)
1. 문제 발생
위와 같은 데이터프레임이 있다.
이 데이터프레임에서 ‘country’ 열의 값이 ‘IT’ 이거나 ‘JP’인 행을 뽑고싶다.
어떻게 해야할까?
# 예제 데이터프레임
df_clean=pd.DataFrame({'country': ['IT','JP','KR','FR']
, 'income':[100,200,300,400]})
2. 문제 해결
크게 2가지 방법이 있다.
첫번째 방법은 or 을 사용하는 것이다.
df_clean[(df_clean['country']=="IT") | (df_clean['country']=="JP")]
또다른 방법은 isin()을 이용하는 것이다.
value_list=["IT","JP"]
df_clean[df_clean['country'].isin(value_list)]
isin( )은 output으로 boolean(True,False) 값을 반환한다.
따라서 위와 같이 인덱싱이 가능하다.
3. 배운 점 및 피드백
- 어러 값을 가지고 있는 리스트가 있을 때, isin( )을 활용하면 효율적인 코딩이 가능하다.
