python - str。包含在pandas数据框中创建新列

Question

我正在探索泰坦尼克号数据集，并希望创建一个名称相似的列。例如，任何包含“Charles”的名称都将显示为“ch”，因为我想稍后使用它们进行分组。
我使用以下代码创建了一个函数:

def cont(Name):
    for a in Name:
        if a.str.contains('Charles'):
            return('Ch')

然后使用以下方法进行应用:

titanic['namest']=titanic['Name'].apply(cont,axis=1)

错误: 'str' object has no attribute 'str' notebook_link

Answer 1

您可以使用向量化的apply返回 bool 值掩码并将满足条件的所有行设置为所需的值，而不是使用循环或str.contains:

titanic.loc[titanic['Name'].str.contains('Charles'), 'namest'] = 'Ch'