Python Pandas自定义体积计算函数else分支未生效问题排查

你代码中对df['volume']的赋值是整列覆盖操作，并非针对循环下标i对应的单行记录赋值：

• 每次触发if分支时，所有行的volume列都会被统一覆盖为x*y*z的计算结果
• 每次触发else分支时，所有行的volume列都会被统一覆盖为8
• 循环运行到最后一行时，如果最后一条记录的depth>60触发if分支，整列就会被最终覆盖为x*y*z的结果，之前else分支写入的8会被全部冲掉。这就是你调试时计数器显示else块确实执行，但最终结果不符合预期的核心原因。

pandas做这类条件赋值优先用向量化运算，不要写原生for循环，代码更简洁、运行效率比逐行循环高几个量级，两种常用写法如下：

写法1：用loc条件筛选赋值 #

def volume(df):
    # 先给所有行默认赋值8
    df['volume'] = 8
    # 仅筛选depth>60的行，覆盖计算体积
    mask = df['depth'] > 60
    df.loc[mask, 'volume'] = df.loc[mask, 'x'] * df.loc[mask, 'y'] * df.loc[mask, 'z']
    return df

写法2：用numpy.where快速实现 #

import numpy as np

def volume(df):
    df['volume'] = np.where(
        df['depth'] > 60,
        df['x'] * df['y'] * df['z'],
        8
    )
    return df

不推荐继续使用逐行for循环的写法：如果数据集行数较多，循环运行速度会极慢。如果一定要写循环，必须指定行下标做单值赋值，不能直接操作整列，且要避免链式索引触发警告：

def volume(df):
    df['volume'] = 0
    for i in range(len(df)):
        if df['depth'].iloc[i] > 60:
            df['volume'].iloc[i] = df['x'].iloc[i] * df['y'].iloc[i] * df['z'].iloc[i]
        else:
            df['volume'].iloc[i] = 8
    return df

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