使用函数根据另一列的值创建 Pandas 列

Question

我想根据数据框中的头衔来识别医生，并创建一个新列来指示他们是否是医生，但我正在为我的代码苦苦挣扎。

doctorcriteria = ['Dr', 'dr']

def doctor(x):
  if doctorcriteria in x:
    return 'Doctor'
  else:
    return 'Not a doctor'

df['doctorcall'] = df.caller_name
df.doctorcall.fillna('Not a doctor', inplace=True)
df.doctorcall = df.doctorcall.apply(doctor)

Answer 1

要使用函数创建新列，您可以使用 apply：

df = pd.DataFrame({'Title':['Dr', 'dr', 'Mr'],
               'Name':['John', 'Jim', 'Jason']})

doctorcriteria = ['Dr', 'dr']

def doctor(x):
    if x.Title in doctorcriteria:
        return 'Doctor'
    else: return 'Not a doctor'

df['IsDoctor'] = df.apply(doctor, axis=1)

但更直接的答案是在 Title 列上使用 map。

doctor_titles = {'Dr', 'dr'}

df['IsDoctor'] = df['Title'].map(lambda title: title in doctor_titles)

使用函数根据另一列的值创建 Pandas 列

Creating a Pandas column based on values of another column using function

python

function

calculated-columns

dataframe

pandas