본문 바로가기

Computer Science/파이썬

[데이터 전처리] Pandas isin

Column이 list의 값들을 포함하고 있는 모든 행들을 골라낼 때 주로 쓴다.
df = pd.DataFrame({'A': [1, 2, 3], 'B': ['a', 'b', 'f']})
  A B
0 1 a
1 2 c
2 3 f

 

df[df['A'].isin([1, 3, 12])]
  A B
0 1 a
2 3 f