키워드 추출 방법 궁급합니다!

안녕하세요! 키워드 추출을 위한 코드와 방법을 알려드리겠습니다. 우선, 코드에서 KEYWORDS는 추출하고 싶은 키워드들의 딕셔너리 형태로 정의되어 있습니다. 그리고 reports 변수는 CONCLUSION 컬럼의 값을 리스트 형태로 저장하고 있습니다. 아래는 키워드가 CONCLUSION 컬럼에 존재하면 해당 row의 관상동맥언급 컬럼을 1로 설정하는 코드입니다. 이 코드를 실행하면 '관상동맥언급' 컬럼에 0 또는 1이 채워진 데이터프레임이 출력됩니다. 0은 해당 row에서 추출하고자 하는 키워드가 존재하지 않은 경우이고, 1은 해당 row에서 추출하고자 하는 키워드가 존재하는 경우입니다. 이 코드를 참고하여 코드를 수정해보시기 바랍니다.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import pandas as pd
import re

df = pd.read_csv('./2015.csv', encoding='utf-8-sig')
df = df.apply(lambda x: x.astype(str))

KEYWORDS = {'CAC': ['coronary calcification', 'coronary artery calcification'],
            'Minimal CAC': ['minimal'],
            'Mild CAC': ['mild'],
            'mild to moderate': [],
            'Moderate CAC': ['moderate'],
            'moderate to severe': [],
            'Severe CAC': ['severe'],
            'stent': [],
            'LAD': [],
            'LM': [],
            'LCX': [],
            'RCA': []}

df['관상동맥언급'] = 0  # 초기값 0으로 설정

for keyword, variants in KEYWORDS.items():
    for variant in variants:
        # keyword가 CONCLUSION 컬럼에 존재하는 경우 해당 row의 관상동맥언급 컬럼을 1로 설정
        df.loc[df['CONCLUSION'].str.contains(variant), '관상동맥언급'] = 1

# 결과 출력
print(df)