Pandas计算净收入代码仅遍历首值的问题排查与修正方法

原代码的问题 #

1. 整列覆盖而非逐行计算：循环里每次判断后直接给df['net_revenue']整列赋值，每一次循环都会把之前的结果全部覆盖，最终整个列只会保留最后一次循环的计算结果，根本没做到逐行处理。
2. 运算符混用错误：第三个条件里用了位运算符&，单个数值判断应该用逻辑运算符and，虽然这里不会直接报错，但逻辑上不规范。
3. Pandas使用思路错误：Pandas的优势是矢量化操作，用for循环逐行处理不仅效率低，还容易出现这类赋值错误。

修复方案 #

方案1：矢量化条件判断（推荐，效率最高） #

用numpy.select批量处理所有行，这是Pandas里处理多条件分支的最优方式：

import numpy as np

# 定义条件列表
conditions = [
    df['discounted_price'] > 600,
    (df['discounted_price'] <= 600) & (df['discounted_price'] > 350),
    (df['discounted_price'] <= 350) & (df['discounted_price'] > 100)
]
# 对应条件的计算规则
choices = [
    df['discounted_price'] * 0.25,
    df['discounted_price'] * 0.15,
    df['discounted_price'] * 0.10
]

# 应用条件，默认处理剩下的情况（<=100）
df['net_revenue'] = np.select(conditions, choices, default=df['discounted_price'] * 0.50)

print(df['net_revenue'])

方案2：apply逐行处理（适合简单逻辑） #

如果逻辑更复杂，也可以用apply逐行调用函数计算：

def calc_net(price):
    if price > 600:
        return price * 0.25
    elif 350 < price <= 600:
        return price * 0.15
    elif 100 < price <= 350:
        return price * 0.10
    else:
        return price * 0.50

df['net_revenue'] = df['discounted_price'].apply(calc_net)

print(df['net_revenue'])

关于“仅遍历第一个值”的说明 #

原代码其实会遍历所有值，但每一次遍历都会覆盖整列的结果。如果看起来只生效了第一个值，可能是后续循环的条件和第一个值的条件一致，或者代码执行时出现了其他异常，但核心问题还是错误地整列赋值，没有逐行计算。

内容的提问来源于stack exchange，提问作者Ranju