在BigQuery中为数据集插入未来7天日期并均分分配至各行

要把生成的7天日期均等分配到原数据集的每一行，核心是通过序号分组逻辑让日期循环匹配行数据，保证每个日期对应的行数尽可能一致。具体实现步骤如下：

• 生成带索引的日期列表：给你已生成的7天日期添加1到7的索引，方便后续关联。
• 给原数据添加行序号：用窗口函数给原数据集的每行生成唯一递增序号。
• 按序号循环匹配日期：通过序号的取余运算，让行数据循环对应7个日期，实现均等分配。

完整代码示例 #

假设你的原数据集表名为your_dataset.your_table，可直接套用以下BigQuery SQL代码：

WITH 
-- 生成带索引的未来7天日期
date_list AS (
    SELECT 
        invite_date,
        ROW_NUMBER() OVER(ORDER BY invite_date) AS date_index
    FROM UNNEST(
        GENERATE_DATE_ARRAY(CURRENT_DATE(), DATE_ADD(CURRENT_DATE(), INTERVAL 7 DAY), INTERVAL 1 DAY)
    ) AS invite_date
),
-- 给原数据添加行序号，并计算对应的日期索引
original_data_ranked AS (
    SELECT 
        *,
        ROW_NUMBER() OVER() AS row_num,
        -- 通过取余实现日期索引的循环匹配
        MOD(row_num - 1, 7) + 1 AS target_date_index
    FROM your_dataset.your_table
)
-- 关联原数据与日期列表，完成日期分配
SELECT 
    odr.*,
    dl.invite_date
FROM original_data_ranked odr
JOIN date_list dl ON odr.target_date_index = dl.date_index
ORDER BY odr.row_num;

关键逻辑说明 #

• MOD(row_num - 1, 7) + 1：这个表达式让行序号从1开始循环对应1-7的日期索引，比如第1-7行对应日期1-7，第8-14行再次循环匹配日期1-7，以此类推。
• 如果原数据总行数无法被7整除（比如共8行），前N个日期会多分配1行（N为总行数除以7的余数），最终实现最接近均等的分配效果。

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