[Python] Pandas isin 구문

 

Python에서 isin 구문은 열이 list의 값을 포함하고 있는 행을 찾을 때 사용

 

인덱스기준으로도 사용이 가능하다

 

예시

import pandas as pd

test = pd.DataFrame({'A': [1, 2, 3, 0], 'B': ['a', 'b', 'f', 'e']})

index기준으로 0과 3에 해당하는 컬럼만 추출하고싶은경우

index_test= [0,3] #인덱스 리스트
test[test.index.isin(index_test)]

 

인덱스가 0과3이 아닌것을 추출하고싶은 경우

index_test= [0,3]
test[~test.index.isin(index_test)]

 

특정 컬럼기준으로 추출하고싶은 경우

a_1= [1]
test[test['A'].isin(a_1)]

 

^^