如何从python中的列中提取字符串

Question

我有一列包含客户的姓名。 现在，我想创建另一个必须只包含以下标题的列： ((先生),(夫人),(博士))

我的专栏是这样的：

print(df['Name'])

        Name
0 William, Mr. Henry
1 Mrs Meryam Hinadi
2 Dr. Richard Barodi
3 Yadar Danari Mr.
4 Gadi Dr. Nawsi

我定义了这个函数，但它不能正常工作。我一定是错过了什么。请纠正我。

def get_status(cols):
        name = cols[0]

        if name == name.str.contains('Mrs'):
            return 'Mrs'
        elif name == name.str.contains('Mr.'):
            return 'Mr.'
        elif name == name.str.contains('Dr.'):
            return 'Dr.'
        else:
            return name

df['Titles'] = df['Name'].apply(get_status)

谢谢。

Answer 1

试试这个：

def get_status(cols):
    if ('Mrs') in cols:
        return 'Mrs'
    elif ('Mr.') in cols:
        return 'Mr.'
    elif ('Dr.') in cols:
        return 'Dr.'
    else:
        return cols

df['Titles'] = df['name'].apply(get_status)

输出：

    name               Titles
0   William, Mr. Henry  Mr.
1   Mrs Meryam Hinadi   Mrs
2   Dr. Richard Barodi  Dr.
3   Yadar Danari Mr.    Mr.
4   Gadi Dr. Nawsi      Dr.