获取表中每个k记录的array_agg（id）数组

Question

我想查询一个雪花表，以便将其所有记录都批准成包含ID阵列的行，最多最多是K元素。 例如，如果k = 2，并且该表包含带有IDS 1,2,3,4,5的记录，我想发布一个可以返回的查询[1,2]，[3,4]，[5]（实际批处理标准将不重要）。有可能吗？

Answer 1

需要一个基于伪柱，我们可以进行分组以创建这些集合。
使用row_number（）和做div根据k的值创建组。

以下几个示例

with cte(col1) as (
    select * from values
    (1),(2),(3),(4),(5),(6),(7),(8),(9)
)
,cte_1 as (
    select ceil(row_number() over (order by null)/3) rn, 
       col1 
    from cte
)
select array_agg(col1) agg_output 
from cte_1 
group by rn order by rn;

-Agg_output
[1，2，3]
[4，5，6]
[7，8，9]

Answer 2

我喜欢Pankaj的答案，我和以为我可以摆脱第二个CTE（而不会使它成为子选择），但最终出于范围的原因而需要。

我确实翻转了使用变量来控制桶。我首先将两个顺序添加到了桶中，我使用了所讨论的值，因此可以看到如何不随机将值分配到存储桶中。其次是数组上的组内部的一个，因此始终“按顺序”始终不需要“按顺序”。但也不需要说明。但是，如果总是需要答案，则需要按顺序进行。

set width = 4;

with table_of_numbers(val) as (
    -- just a CTE to make 14 numbers
    select 
        row_number() over(order by null)
    from table(generator(ROWCOUNT => 14))
), pre_cond as(
    select *,
        ceil(row_number() over (order by val)/$width) as rn
    from table_of_numbers
)
select 
    array_agg(val)within group(order by val) agg_output 
from pre_cond 
group by rn order by rn;

给出：

agg_output
[1，2，3，4] [5，6，7，8
]
[9，10，11，12 ]
[13，14]

在存储桶上注意的点是Row_number在1开始值将是自然间隔，因此需要CEIL将PRIO N-1部分推入同一桶。由于不只有整数划分（自然地板）减去1 form_number，因此无法利用更简单的逻辑。

Answer 3

这可能不会胜过pankaj的答案，但避免了子查询。也许array_slice读取更多的自我记录。在组中添加（订购。

set K=3;

select array_slice(array_agg(col) over (),
                   sum($K) over (order by col) - $K,  
                   sum($K) over (order by col)) as array_part
from cte
qualify array_size(array_part) > 0