Pandas - 在列 A 中搜索字符串并将其添加到新列 B

Question

鉴于我有一个来自“train”数据集的“Name”列，我想从中提取标题，将其添加到新创建的“Title”列中。

我想过使用re.search("([A-Za-z]+)\\.")但我不确定如何遍历整个列并将其传递给新列。

预先感谢您的帮助。

Answer 1

您可以使用以下代码：

import pandas as pd

df=pd.DataFrame([['Braund, Mr. Owen Harris'],['Cumings, Mrs. John Bradley'],
            ['Heikkinen, Miss. Laina'],['Futrellem, Mrs. Jacques Heath'],
            ['Allen, Mr. William Henry']],
            columns=['Name'])

df['title']=df['Name'].apply(lambda s:s.split(',')[1].split('.')[0])

Answer 2

尝试这个

df - 带有列名称的数据框

df["Title"] = df["Name"].apply(lambda name: re.search(name, "([A-Za-z]+).").group(0))

它将在您的 df 中创建新的列标题