python - 熊猫应用功能不会为列分配值

Question

我试图把这个逻辑放在熊猫数据框上

IF base_total_price > 0
    IF base_total_discount = 0
        actual_price = base_total_price
   
     IF base_total_discount > 0
            actual_price = base_total_price +base_total_discount

IF base_total_price = 0
     IF base_total_discount > 0
          actual_price = base_total_discount
     IF base_total_discount = 0
          actual_price = 0

所以我写了这两个应用函数

#for all entries where base_total_price > 0
df_slice_1['actual_price'] = df_slice_1['base_total_discount'].apply(lambda x: df_slice_1['base_total_price'] if x == 0 else df_slice_1['base_total_price']+df_slice_1['base_total_discount'])

#for all entries where base_total_price = 0
df_slice_1['actual_price'] = df_slice_1['base_total_discount'].apply(lambda x: x if x == 0 else df_slice_1['base_total_discount'])

当我运行代码时出现此错误

ValueError: Wrong number of items passed 20, placement implies 1

我知道它正在尝试将更多值放在一列中，但我不明白为什么会发生这种情况或如何解决这个问题。我需要做的就是使用新列 `actual_price` 更新数据框，并且我需要根据上述逻辑计算该列的值。请建议我一种更好的实现逻辑的方法或纠正我

标签： pythonpandasdataframejupyter

Answer 1

样本数据会很有用。请尝试使用np.select(condtions, choices)

 Conditions=[(df.base_total_price > 0)&(df.base_total_discount == 0),(df.base_total_price > 0)&(df.base_total_discount > 0),\
    (df.base_total_price == 0)&(df.base_total_discount > 0),\
    (df.base_total_price == 0)&(df.base_total_discount == 0)]
choices=[df.base_total_price,df.base_total_price.add(df.base_total_discount),df.base_total_discount,0]
df.actual_price =np.select(Conditions,choices)