Python Dataframe 根据其中一列判断 来增减字符

Question

我有一个Dataframe
df = pd.DataFrame({'Category':['A','C','D','C,D','C,D','A,B,C','A,D','C','C','C','C','C','A,D','A,B,C','A,B,C','A,B,C','D','D','D','D','D'],
'Logic':['Add C','Add D','Add C','','','Del C','Add C','','','Del C','','Add E','Add C','','Upd B To F','','','','','Upd D To F','Add B']})

我想根据‘Logic’这列的提示 对 Category这列进行更改 达到以下结果

请各位帮助 谢谢！

Answer 1

import pandas as pd
df = pd.DataFrame({'Category':['A','C','D','C,D','C,D','A,B,C','A,D','C','C','C','C','C','A,D','A,B,C','A,B,C','A,B,C','D','D','D','D','D'],
'Logic':['Add C','Add D','Add C','','','Del C','Add C','','','Del C','','Add E','Add C','','Upd B To F','','','','','Upd D To F','Add B']})

def judge(rawStr, condition):
    if "Add" in condition:
        addValue = ",".join(i for i in condition.split(' ')[1:])
        newStr = rawStr +','+addValue
        return newStr
    elif "Del" in condition:
        delValue = condition.split(' ')[-1]
        if "," in rawStr:
            rawList = rawStr.split(',')
            rawList.remove(delValue)
            newStr = ",".join(i for i in rawList)
            return newStr
        else: 
            return ""
    elif "Upd" in condition:
        rawV = condition.split(' ')[1]
        newV = condition.split(' ')[3]
        newStr = rawStr.replace(rawV,newV)
        return newStr
    else:
        return rawStr
df['Category_result'] = df.apply(lambda x: judge(x['Category'],x['Logic']),axis=1)