Pandas如何按type列规则新增计数列：遇S计数到L后停止

实现代码 #

import pandas as pd
import numpy as np

df = pd.DataFrame({"Value":[1,4,2,1,2,1,2,2],'type':['S','C','L','C','C','S','L','S']})

# 计算S的累计计数值（基准值）
s_cum = df['type'].eq('S').cumsum()
# 生成有效位置掩码：标记处于S到对应L之间（含两端）的行
valid_mask = (s_cum - df['type'].eq('L').cumsum()).gt(0) | df['type'].eq('L')
# 赋值，无效位置置空
df['exit_value'] = np.where(valid_mask, s_cum, np.nan)
# 可选：转为可空整数类型，显示时空值为空白，和期望结果样式一致
df['exit_value'] = df['exit_value'].astype('Int64')

逻辑说明 #

1. 首先计算s_cum，统计从开头到当前行一共出现了多少次S，这就是你要的计数基准。
2. 然后计算有效位置掩码valid_mask：
   • 用S的累计计数减去L的累计计数，差值大于0时，说明当前行处于「最近一次S出现之后，还没遇到对应的L」的状态
   • 额外匹配type == L的行，保证L行也保留当前计数
3. 最后用np.where给有效行填充基准值，无效行填充空值即可。

运行后得到的df和你给出的期望结果完全一致。

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