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_discountIF 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 > 0df_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 = 0df_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 解决了这个问题。感谢所有回复的人