SQL Server中如何为每个ID符合条件的首行添加FLAG 'Y'

问题描述 #

现有Table A表，数据结构及内容如下：

ID      AMOUNT   Order
 -------------------------
  1       100       1
  1       300       2
  1       320       3
  2       100       1
  2       200       2
  2       423       3
  2       613       4
  3       112       1
  3       218       2
  3       290       3
  3       480       4

表中ID存在重复，每个ID对应的数据按Order字段排序。需求是：仅为每个ID下**第一个满足Order≥3且Amount≥400**的行添加FLAG 'Y'，其余符合条件的行不标记，期望输出如下：

ID      AMOUNT   Order  Flag
 ------------------------------
  1       100       1
  1       300       2
  1       320       3
  2       100       1
  2       200       2
  2       423       3      Y
  2       613       4
  3       112       1
  3       218       2
  3       290       3      
  3       480       4      Y

原SQL语句会标记所有符合条件的行，无法实现仅标记首行的需求：

Select * , case when (Order >= 3 and Amount >=400) then 'Y' else '' end as FLAG from TableA

解决方案 #

可以通过窗口函数计算每个ID下符合条件的最小Order值，再匹配判断实现需求，以下是简洁高效的SQL语句（适用于MySQL 8+、PostgreSQL、SQL Server、Oracle等支持窗口函数的数据库）：

SELECT
    ID,
    AMOUNT,
    "Order",
    CASE
        WHEN "Order" = MIN(CASE WHEN "Order" >= 3 AND AMOUNT >= 400 THEN "Order" END) OVER (PARTITION BY ID)
             AND "Order" >= 3 AND AMOUNT >= 400 THEN 'Y'
        ELSE ''
    END AS Flag
FROM TableA
ORDER BY ID, "Order";

逻辑说明 #

1. 计算每个ID的目标Order值：
   MIN(CASE WHEN "Order" >=3 AND AMOUNT >=400 THEN "Order" END) OVER (PARTITION BY ID)
   • 针对每个ID分组，筛选出满足Order≥3且Amount≥400的行，取其中最小的Order值；若无符合条件的行，该值为NULL。
2. 判断标记FLAG：
   外层CASE判断当前行是否同时满足两个条件：
   • 自身符合Order≥3且Amount≥400；
   • 当前行的Order等于该ID下的目标最小Order值。
     满足则标记'Y'，否则为空字符串。

这种写法无需子查询或表连接，直接通过窗口函数完成计算，性能更优，且逻辑清晰易懂。

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