dataframe 칼럼과 list를 비교하여 데이터 추출

일전에 사용했던 칼럼에 특정 단어가 포함 되어있는 지를 검증하는 코드를 활용했다.

new_df = df.loc[df['칼럼1'].str.contains('단어1|단어2')]

검증하고싶은 list를 위의 contains안에 들어가도록 형태를 바꿔야 한다.

 

my_list = ['단어1', '단어2', '단어3']

# my_list를 포함하는 경우
new_df = df.loc[df['칼럼1'].str.contains("|".join(my_list))]

# my_list를 포함하지 않는 경우
new_df2 = df.loc[~df['칼럼1'].str.contains("|".join(my_list))]

끝