熊猫根据列表从列中删除部分文本

时间:2021-04-29 19:26:27

标签: python pandas

我有一列包含比赛标题和体重等级列表。我需要根据 Weight_Class 列表搜索 Fight_Name 列,并在我的 df 中填充一列,如下所示:

编辑澄清:我需要这样的输出。我只有一个重量等级列表,我需要将它设为我的 df 中的一列,该列根据 Fight_Name 列进行匹配

<头>
Fight_Name Weight_Class
中量级比赛 中量级
UFC 临时中量级冠军争夺战 中量级
UFC 轻重量级冠军争夺战 轻量级
Strikeforce 原子量级冠军争夺战 原子量级

1 个答案:

答案 0 :(得分:1)

你想要这个吗?

Note - 将 for 循环中的 weight_class_list 替换为您的列表名称 -

def extract(x):
    for item in weight_class_list:
        if item in x:
            return item
        
df['Weight_Class'] = df['Fight_Name'].apply(lambda x: extract(x))