TSQL计数连续记录

时间:2016-01-20 09:52:21

标签: sql-server tsql grouping partition

假设我有以下记录:

KeyCol     ColA     ColB
------------------------
1          1        A
2          2        B
3          2        B
4          2        C
5          2        B
6          1        A
7          2        B
8          2        B

我希望使用此结果计算ColA和ColB中具有相同值的连续记录

Col A      ColB     Start   Count
---------------------------------
1          A        1       1
2          B        2       2
2          C        4       1
2          B        5       1
1          A        6       1
2          B        7       2

关于分组和计数有很多类似的问题,但我没有看到如何将它翻译成这个。特别是许多其他示例没有明确的键列。

我尝试使用PARTITION函数计算连续记录的数量并从那里取出:

SELECT KeyCol, ColA, ColB
      ,ROW_NUMBER() OVER 
            (   PARTITION
                BY ColA, ColB
                ORDER BY KeyCol
            ) as RowNo
FROM MyTable

但是,这会产生这样的结果:

KeyCol    Col A      ColB     RowNo
---------------------------------
1         1          A        1
2         2          B        1
3         2          B        2
4         2          C        1
5         2          B        3   (Needs to be 1)
6         1          A        2   (Needs to be 1)
7         2          B        4   (Needs to be 1)
8         2          B        5   (Needs to be 2)

如您所见,即使记录不连续,所有相同的ColA,ColB的行号也会增加。

非常感谢!

1 个答案:

答案 0 :(得分:5)

这是Gaps and Islands problem。您需要使用排名函数来识别ColB的相同值的组(岛)。以下查询:

SELECT  KeyCol,
        ColA,
        ColB,
        GroupBy = ROW_NUMBER() OVER(ORDER BY KeyCol) - 
                    ROW_NUMBER() OVER(PARTITION BY ColA, ColB ORDER BY KeyCol)
FROM    dbo.T
ORDER BY KeyCol;

您将获得输出:

KeyCol     ColA     ColB    GroupBy
-----------------------------------------
1          1        A           0
2          2        B           1
3          2        B           1
4          2        C           3
5          2        B           2
6          1        A           4
7          2        B           3
8          2        B           3   

如您所见,这标识了您的岛屿,其中两个(或更多)连续行具有相同的ColA和ColB值,您将在GroupBy列中获得相同的值。

一旦你有了这个,这是一个简单的分组,通过它来获得你所需的输出。给出最终查询(带有样本数据):

DECLARE @T TABLE (KeyCol INT, ColA INT, ColB CHAR(1));
INSERT @T (KeyCol, ColA, ColB)
VALUES
    (1, 1, 'A'), (2, 2, 'B'), (3, 2, 'B'), (4, 2, 'C'),
    (5, 2, 'B'), (6, 1, 'A'), (7, 2, 'B'), (8, 2, 'B');

WITH RankedData AS
(   SELECT  KeyCol,
            ColA,
            ColB,
            GroupBy = ROW_NUMBER() OVER(ORDER BY KeyCol) - 
                        ROW_NUMBER() OVER(PARTITION BY ColA, ColB ORDER BY KeyCol)
    FROM    @T
)
SELECT  ColA, 
        ColB,
        Start = MIN(KeyCol),
        [Count] = COUNT(*)
FROM    RankedData
GROUP BY ColA, ColB, GroupBy
ORDER BY Start;