LAG 和 LEAD

Snowflake 中的 Window Functions

Jake Roach

Field Data Engineer

LAG

LAG 用于将某值与"上一条记录"中的值进行比较

SELECT
    <fields>,


    LAG(<1>, <2>, <3>) OVER(
        PARTITION BY <4>
        ORDER BY <5>
    )


...;

<1>：要获取的上一条记录中的字段

<2>：向后回看的记录数

<3>：记录不存在时的默认值，0

<4>：用于分区的字段

<5>：用于确定记录顺序的字段

LAG

使用 LAG 生成的数据集示例

LAG

SELECT
    member_id AS m_id, 
    workout_date AS wd,
    calories_burned AS cb,

    -- 取上一次锻炼消耗的卡路里
    -- （即上一条记录）

    LAG(calories_burned, 1) OVER(
        PARTITION BY member_id
        ORDER BY workout_date
    ) AS past_cb,


FROM fitness.workouts;

  m_id  |       wd     |   cb  |  past_cb 
 ------ | ------------ | ----- | ---------
 m_192  |  2024-01-01  |  105  |   null
 m_192  |  2024-01-03  |  156  |   105
 m_192  |  2024-01-04  |   69  |   156
 m_192  |  2024-01-10  |  102  |    69

 m_74   |  2024-02-10  |  374  |   null
 m_74   |  2024-02-13  |  396  |   374
 m_74   |  2024-02-14  |  504  |   396

 m_233  |  2024-03-05  |  51   |   null 
 m_233  |  2024-03-12  |  81   |    51

LAG

SELECT
    ...
    LAG(calories_burned, 1) OVER(
        PARTITION BY member_id
        ORDER BY workout_date
    ) AS past_cb,

    -- 计算消耗卡路里的差值
``` {sql}
    calories_burned - LAG(calories_burned, 1, calories_burned) OVER(
        PARTITION BY member_id
        ORDER BY workout_date
    ) AS more_cb,


FROM fitness.workouts;

LAG

                  m_id  |       wd     |   cb  |  past_cb  |  more_cb
                ------- | ------------ | ----- | --------- | ---------
                 m_192  |  2024-01-01  |  105  |    null   |      0
                 m_192  |  2024-01-03  |  156  |    105    |     51
                 m_192  |  2024-01-04  |   69  |    156    |    -87
                 m_192  |  2024-01-10  |  102  |     69    |     33

                 m_74   |  2024-02-10  |  374  |    null   |      0
                 m_74   |  2024-02-13  |  396  |    374    |     22
                 m_74   |  2024-02-14  |  504  |    396    |    108

                 m_233  |  2024-03-05  |  51   |    null   |      0
                 m_233  |  2024-03-12  |  81   |     51    |     30

LEAD

LEAD 用于将某值与"未来"记录中的值进行比较

<1>：要获取的后续记录中的字段

<2>：向前查看的记录数

<3>：记录不存在时的默认值

<4>：用于分区的字段

<5>：用于确定记录顺序的字段

SELECT
    <fields>,


    LEAD(<1>, <2>, <3>) OVER(
        PARTITION BY <4>
        ORDER BY <5>
    )


...;

常用于预测类任务

LEAD

SELECT
    member_id, 
    workout_date
    calories_burned,

    -- 本次锻炼后，找到下一次锻炼日期
    LEAD(workout_date, 1) OVER(
        PARTITION BY member_id
        ORDER BY workout_date
    ) AS next_workout_date

FROM fitness.workouts;

LEAD

            m_id  |  workout_date  |  calories_bured  |  next_workout_date
          ------- | -------------- | ---------------- | -------------------
           m_192  |   2024-01-01   |        105       |     2024-01-03
           m_192  |   2024-01-03   |        156       |     2024-01-04
           m_192  |   2024-01-04   |         69       |     2024-01-10
           m_192  |   2024-01-10   |        102       |         null

           m_74   |   2024-02-10   |        374       |     2024-02-13
           m_74   |   2024-02-13   |        396       |     2024-02-14
           m_74   |   2024-02-14   |        504       |         null

           m_233  |   2024-03-05   |         51       |     2024-03-12 
           m_233  |   2024-03-12   |         81       |         null

Passons à la pratique !

Snowflake 中的 Window Functions