python - 如何从 B 列的字符串中删除 A 列的字符串

Question

想知道如果您有两列(A = 'Name'，B = 'Name_Age')，是否有一种快速方法可以从 'Name_Age' 中删除 'Name'，以便您可以快速获取 'Age'，就像反转一样级联？？

我考虑过“字符串分割”，但在某些情况下(当没有字符串分割因子时)我确实需要一种方法来从另一列的字符串中删除一列的字符串。

#example data below:

import pandas as pd

data =  {'Name':['Mark','Matt','Michael'], 'Name_Age':['Mark 14','Matt 29','Michael 18']}

df = pd.DataFrame(data)

Answer 1

您可以尝试使用 pandas apply 函数，它允许您定义自己的函数以传递到数据帧的每一行:

def age_from_name_age(name, name_age):
    return name_age.replace(name, '').strip()

df['Age'] = df.apply(lambda x: age_from_name_age(x['Name'], x['Name_Age']),
                     axis='columns')

age_from_name_age 采用两个字符串(名称和 name_age)，并仅返回年龄。然后，在 apply 语句中，我定义一个匿名 lambda 函数，该函数仅接受一行并将正确的字段传递给 age_from_name_age。