如何在pandas中使用np.where结合上一行值创建条件新列及解决KeyError报错

你碰到的问题主要来自三个层面的错位：

• 列引用不存在：当前df只有Count、Status两列，代码里直接在np.where中引用还没创建的Previous列，必然触发KeyError
• 偏移方向完全搞反：你提到的Excel公式=IF(B2="new",A2,C3)本身就有行引用方向错误——在C2单元格要实现“非new行取上一行C值”的逻辑，应该引用C1（上一行）而非C3（下一行）；对应到pandas里，取上一行的值要用shift(1)，你写的shift(-1)是取下方行的值，和需求方向完全相悖
• 场景匹配错误：你要的效果本质是「遇到Status为new的行，就把当前Count值作为基准，后续非new行一直沿用这个基准值，直到下一个new行出现」，这是pandas中典型的前向填充场景，存在逐行的循环依赖，没法靠单轮np.where加shift一次性完成计算。

哪怕是几十万行的数据，这套写法的运行效率也足够高，完全复现Excel的计算逻辑：

1. 先初始化输出列C：只保留Status为new的行对应的Count值，非new行先留空

   df['C'] = df.loc[df['Status'] == 'new', 'Count']
   

2. 对C列做前向填充，把上方最近的非空值向下补全到所有空行

   df['C'] = df['C'].ffill()
   

用一组小样本测试，原始输入数据：

跑完上述代码后，C列的输出结果和手动在Excel里拉公式的结果完全一致：

如果你的Excel里第一行C列本身有固定初始值，只需要在调用ffill()之前，手动给C列的第一行赋值对应的初始值，就能1:1复现Excel的计算结果。如果最开头几行的Status都不是new，计算出来的C列会是空值，这个行为也和Excel完全一致——因为初始位置没有可以沿用的基准值。

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