无需迭代的Pandas列向前填充：ExtendsValid规则高效实现方案

直接利用Pandas的向量化操作结合ffill()，无需分组或迭代，代码简洁高效：

import numpy as np
import pandas as pd

# 构造辅助列：Valid=1标记为1，ExtendsValid=0标记为0，其余为NaN
df['_helper'] = np.where(df['Valid'] == 1, 1, 
                         np.where(df['ExtendsValid'] == 0, 0, np.nan))

# 向前填充+补0得到FinalValid
df['FinalValid'] = df['_helper'].ffill().fillna(0).astype(int)

# 可选：删除辅助列
df.drop('_helper', axis=1, inplace=True)

逻辑解释 #

1. 辅助列构造：

   • 把原生有效行（Valid=1）标记为1
   • 把有效性终止行（ExtendsValid=0）标记为0
   • 其他行（待判断的ExtendsValid=1行）设为NaN

2. 向前填充（ffill）：

   • 从Valid=1的位置开始，后续所有NaN（即ExtendsValid=1的行）会被向前填充为1，直到遇到下一个0（ExtendsValid=0）
   • 所有Valid=1之前的NaN行，因为没有前置有效行，最终会被fillna(0)设为0

示例验证 #

假设原始DataFrame如下：

处理后FinalValid结果：

完全符合需求：只有前置存在有效行（原生或扩展）的ExtendsValid=1行才会被标记为有效，有效状态会持续到ExtendsValid=0终止。

内容的提问来源于stack exchange，提问作者DISC-O