通过Pandas如何实现对Excel表格其中一列(例如A列),对A列进行筛选,筛选出包含某特定字符,然后新增一列,将符合特定条件的赋值新值给B列

想实现这么一个功能见下图:

img

对A列进行筛选,同时新增B列
如果A列包含京东,B列赋值京东
如果A列包含阿里,B列赋值阿里
如果A列包含淘宝,B列赋值淘宝
其余剩下的,B列填写其他

不指定应该用什么函数,或者方法, 恳请指点!谢谢!

读取excel的代码就省略了,原理就是把“A列”抽出来使用series的map方法,生成新的一列数据,再添加回dataframe,取名“B列”

def fun(s):
    if "京东" in s:
        return "京东"
    elif "阿里" in s:
        return "阿里"
    elif "淘宝" in s:
        return "淘宝"
    else:
        return "其他"

df['B列'] = df['A列'].map(fun)