Python如何将标记值前n-1行的exclusion_flag设为-1以便后续删除

以下提供两种常用场景的实现逻辑，你可以根据自己的技术栈选择：

Python Pandas 实现 #

如果你的数据是结构化表格，用Pandas处理可以直接参考以下代码：

import pandas as pd

# 步骤1：读入数据，确保行顺序和你实际存储的顺序一致
# 示例数据构造，实际使用时替换为你的读入逻辑，比如pd.read_csv("你的文件路径")
data = [
    ["1BOP2UC-1", 0],
    ["1BOP2UC-2", 0],
    ["1BOP2UC-3", 0],
    ["1BOP2UC-4", 4],
    ["1BOP2UC-5", 0],
    ["1BOP2UC-6", 0],
    ["1BOP2UC-7", 0],
    ["1BOP2UC-8", 2],
    ["1BOP2UC-9", 0],
    ["1BOP2UD-1", 0],
    ["1BOP2UD-2", 0],
    ["1BOP2UD-3", 0],
    ["1BOP2UD-4", 0],
    ["1BOP2UD-5", 4],
]
df = pd.DataFrame(data, columns=["UID", "exclusion_flag"])
df_processed = df.copy()

# 步骤2：如果需要按UID前缀分组处理（即不跨前缀修改行），执行这段逻辑
# 提取UID前缀
df_processed['uid_prefix'] = df_processed['UID'].str.split('-').str[0]
def process_single_group(group):
    group = group.copy()
    # 遍历当前分组内所有标记行
    for idx, row in group[group['exclusion_flag'] > 0].iterrows():
        n = row['exclusion_flag']
        group_first_idx = group.index[0]
        # 计算修改范围的起始索引，避免越界到分组外或者小于0
        start_idx = max(group_first_idx, idx - (n - 1))
        group.loc[start_idx: idx -1, 'exclusion_flag'] = -1
    return group
df_processed = df_processed.groupby('uid_prefix', group_keys=False).apply(process_single_group)
# 删除临时生成的前缀列
df_processed = df_processed.drop(columns=['uid_prefix'])

# 如果你不需要按UID前缀分组，直接全局处理的话，替换步骤2为以下代码即可
"""
for idx, row in df_processed[df_processed['exclusion_flag'] > 0].iterrows():
    n = row['exclusion_flag']
    start_idx = max(0, idx - (n -1))
    df_processed.loc[start_idx: idx-1, 'exclusion_flag'] = -1
"""

# 输出结果验证
print(df_processed)

运行后输出的结果和你给出的预期输出完全一致。

SQL 实现 #

如果你的数据存在关系型数据库中，可以参考以下SQL逻辑（需要你的表有自增的连续行号列row_id保证行顺序）：

-- 先找出所有标记行的行号和对应的标记值
WITH flag_positions AS (
    SELECT 
        row_id,
        exclusion_flag AS n
    FROM your_table_name
    WHERE exclusion_flag > 0
)
-- 更新标记行之前的n-1行
UPDATE your_table_name t
SET exclusion_flag = -1
WHERE EXISTS (
    SELECT 1
    FROM flag_positions f
    WHERE t.row_id BETWEEN (f.row_id - f.n + 1) AND (f.row_id - 1)
);

如果需要按UID前缀分组处理，只需要在WHERE条件里加上前缀匹配的逻辑即可。

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