求正确SAS代码：按A列分组保留B列优先级值（红优先无红留蓝）

原代码问题点 #

原代码无法输出ABC13的blue记录，核心原因是：

• 对于无red的分组（如ABC13），keep变量始终为0：first.A时初始化keep=0，后续遇到blue时仅执行keep=keep（值仍为0），遇到green时设为0，最终last.A and keep条件不满足，无法触发输出。
• 逻辑未覆盖「无red时保留blue」的标记需求，仅依赖red触发keep=1，导致无red分组没有输出触发条件。

修正方案（两种实现思路） #

方案1：两步法（先标记分组特征，再筛选）

先提取每个分组是否存在red，再根据标记筛选目标记录：

/* 第一步：生成分组标记，记录每组是否包含red */
data group_flag;
    set have;
    by A;
    retain has_red;
    if first.A then has_red = 0;
    if B = "red" then has_red = 1;
    if last.A then output;
    keep A has_red;
run;

/* 第二步：根据分组标记筛选记录 */
data want;
    merge have group_flag;
    by A;
    if has_red then do;
        /* 有red的分组，仅保留第一条red */
        if B = "red" and first.B then output;
    end;
    else do;
        /* 无red的分组，仅保留第一条blue */
        if B = "blue" and first.B then output;
    end;
run;

方案2：单数据步完成逻辑

通过retain变量同时标记「是否有red」和「是否已输出blue」，在一次遍历中完成筛选：

data want;
    set have;
    by A;
    retain has_red found_blue;
    /* 分组初始化 */
    if first.A then do;
        has_red = 0;
        found_blue = 0;
    end;

    /* 标记当前分组是否存在red */
    if B = "red" then has_red = 1;

    /* 执行输出逻辑 */
    if has_red then do;
        /* 有red则输出第一条red */
        if B = "red" and first.B then output;
    end;
    else do;
        /* 无red则输出第一条blue，避免重复输出 */
        if B = "blue" and not found_blue then do;
            output;
            found_blue = 1;
        end;
    end;

    drop has_red found_blue;
run;

效果验证 #

两种方案均可得到期望输出：

A       B
ABC12   red
ABC13   blue

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