在Python中定位子字符串并将其提取到新列

Question

我有一个数据框，想要在该列中找到一个关键术语。如果“Jobtitle”列中的行包含该术语，我希望它提取该术语并将其返回到新列。我尝试使用以下代码，但到目前为止返回错误

if 'engineer' in glassdoor['Jobtitle']:

    glassdoor['job_cat'] = glassdoor['Jobtitle'].str.extract('(engineer)', expand=True)

    glassdoor['job_cat']

else:

    print('error')

Answer 1

您可以尝试使用np.where+containsglassdoor['job_cat'] = np.where(glassdoor['Jobtitle'].str.contains('engineer'), 'engineer', 'nothing')