Pandas apply 函数不会为列赋值

Question

我正在尝试将此逻辑放在 pandas 数据框上

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”更新数据框，我需要根据上述逻辑计算该列的值。请建议我更好的实现逻辑的方法或纠正我

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)

Answer 2

我只是通过使用 iterrows 解决了这个问题。感谢所有回复的人