BigQuery SQL for 28-day sliding window aggregate (without writing 28 lines of SQL)

前端 未结 3 624
不思量自难忘°
不思量自难忘° 2020-12-01 05:33

I\'m trying to compute a 28 day moving sum in BigQuery using the LAG function.

The top answer to this question

Bigquery SQL for sliding window aggregate

相关标签:
3条回答
  • 2020-12-01 06:13

    The BigQuery documentation doesn't do a good job of explaining the complexity of window functions that the tool supports because it doesn't specify what expressions can appear after ROWS or RANGE. It actually supports the SQL 2003 standard for window functions, which you can find documented other places on the web, such as here.

    That means you can get the effect you want with a single window function. The range is 27 because it's how many rows before the current one to include in the sum.

    SELECT spend,
           SUM(spend) OVER (PARTITION BY user ORDER BY date ROWS BETWEEN 27 PRECEDING AND CURRENT ROW),
           user,
           date
    FROM user_spend;
    

    A RANGE bound can also be extremely useful. If your table was missing dates for some user, then 27 PRECEDING rows would go back more than 27 days, but RANGE will produce a window based on the date values themselves. In the following query, the date field is a BigQuery TIMESTAMP and the range is specified in microseconds. I'd advise that whenever you do date math like this in BigQuery, you test it thoroughly to make sure it's giving you the expected answer.

    SELECT spend,
           SUM(spend) OVER (PARTITION BY user ORDER BY date RANGE BETWEEN 27 * 24 * 60 * 60 * 1000000 PRECEDING AND CURRENT ROW),
           user,
           date
    FROM user_spend;
    
    0 讨论(0)
  • 2020-12-01 06:32

    Here is an alternate take that I found to be flexible and effective:

    WITH users AS
     (SELECT 'Isabella' as user, 1 as spend, DATE(2020, 03, 28) as date
      UNION ALL SELECT 'Isabella', 2, DATE(2020, 03, 29)
      UNION ALL SELECT 'Daniel', 3, DATE(2020, 03, 24)
      UNION ALL SELECT 'Andrew', 4, DATE(2020, 03, 23)
      UNION ALL SELECT 'Daniel', 5, DATE(2020, 03, 11)
      UNION ALL SELECT 'Jose', 6, DATE(2020, 03, 17))
    SELECT 
    user,
    max(sum(case date_diff(date(2020,04,15), date, day) between 0 and 28
            when true then spend else 0 end)) over(partition by user) as spend_28_day_sum
    FROM users
    group by user
    
    +------------------------------+
    | user      | spend_28_day_sum |
    +------------------------------+
    | Andrew    | 4                |
    | Daniel    | 3                |
    | Isabella  | 3                |
    | Jose      | 0                |
    +------------------------------+
    

    You could change the specified date for the "window function" to current_date() or cross join with a generated date array to see how users change over time.

    0 讨论(0)
  • 2020-12-01 06:40

    Bigquery: How to get a rolling time range in a window clause.....

    This is an old post, but I spend a long time searching for a solution, and this post came up so maybe this will help someone.

    IF your partition of your window clause does not have a record for every day, you need to use the RANGE clause to accurately get a rolling time range, (ROWS would search the number records, which would to go too far back since you don't have a record for every day in your PARTITION BY). The problem is that in Bigquery RANGE clause does not support Dates.

    From BigQuery's documentation:

    numeric_expression must have numeric type. DATE and TIMESTAMP are not currently supported. In addition, the numeric_expression must be a constant, non-negative integer or a parameter.

    The workaround I found was to use the UNIX_DATE(date_expression) in the ORDER BY clause along with a RANGE clause:

    SUM(value) OVER (PARTITION BY Column1 ORDER BY UNIX_DATE(Date) RANGE BETWEEN 5 PRECEDING AND CURRENT ROW

    0 讨论(0)
提交回复
热议问题