파이썬에서 새 열에 하위 문자열 찾기 및 추출

데이터 프레임이 있고 열에서 핵심 용어를 찾고 싶습니다. 'Jobtitle'열의 행에 용어가 포함되어 있으면 추출하여 새 열로 반환합니다. 다음 코드로 시도했지만 지금까지 오류를 반환합니다.

if 'engineer' in glassdoor['Jobtitle']:
    glassdoor['job_cat'] = glassdoor['Jobtitle'].str.extract('(engineer)', expand=True)
    glassdoor['job_cat']
else:
    print('error')

 

해결 방법

 

np.where + contains 로 시도 할 수 있습니다.

glassdoor['job_cat'] = np.where(glassdoor['Jobtitle'].str.contains('engineer'), 'engineer', 'nothing')

 

참조 페이지 https://stackoverflow.com/questions/63759780