콘솔워크

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

프로그래밍/python

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

이휘재123 2022. 9. 19. 15:27
반응형

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

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))]

반응형