如何在SAS中将amount列值添加到对应数组元素中？

假设你已经完成左连接，得到了包含ID、Date、qty_1至qty_5、amount的数据集（比如命名为have_joined），用以下代码就能实现你要的效果：

核心代码 #

data want;
    set have_joined;
    /* 把qty_1到qty_5定义成数组，方便批量处理 */
    array qty[5] qty_1-qty_5;
    /* 遍历数组，找到值为1的那个元素，把amount加进去 */
    do i = 1 to dim(qty);
        if qty[i] = 1 then do;
            qty[i] = qty[i] + amount;
            leave; /* 找到目标就退出循环，提升效率 */
        end;
    end;
    /* 没有匹配到amount的行，把amount设为0，和目标输出一致 */
    if missing(amount) then amount = 0;
run;

代码解释 #

• 数组定义：array qty[5] qty_1-qty_5; 把分散的qty变量打包成数组，避免逐个写变量名的繁琐。
• 匹配累加逻辑：你的原始数据中每行仅有一个qty元素为1，其余为0，循环找到这个1的位置后，直接将amount值累加到该元素上。
• 缺失值处理：左连接后未匹配到的行amount会缺失，这里直接将其设为0，和示例输出对齐。

灵活适配版（qty变量数量不固定时使用） #

如果你的qty变量数量不确定（比如不止5个），可以用下面的代码自动识别所有以qty_开头的变量，无需手动指定数组长度：

data want;
    set have_joined;
    /* 自动匹配所有qty_开头的变量 */
    array qty[*] qty_:;
    do i = 1 to dim(qty);
        if qty[i] = 1 then do;
            /* coalesce函数直接将缺失的amount转为0，一步处理缺失值 */
            qty[i] = qty[i] + coalesce(amount, 0);
            leave;
        end;
    end;
    amount = coalesce(amount, 0);
run;

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