SQL Bigquery：将特定组的选择限制为 10

下面是示例表

目前，该表对于每个 ID 都有无限的条目。

我的要求是，首先按 rand 升序对 ID 进行排序。然后只取前 2 行 ['ID' 和 'companies'.]

CREATE TABLE table_name (
    ID int,
    companies varchar(255),
    rand float(2)
);

INSERT INTO table_name VALUES (1, 'a', 0.2);
INSERT INTO table_name VALUES (1, 'b', 0.6);
INSERT INTO table_name VALUES (2, 'a', 0.4);
INSERT INTO table_name VALUES (2, 'b', 0.5);
INSERT INTO table_name VALUES (2, 'c', 0.3);
INSERT INTO table_name VALUES (3, 'a', 0.6);
INSERT INTO table_name VALUES (3, 'b', 0.7);
INSERT INTO table_name VALUES (3, 'c', 0.4);
INSERT INTO table_name VALUES (3, 'd', 0.2);

即最终表每个 ID 最多只能包含 2 行。 （不需要包含 rand 列）

你想要row_number()?

select * except(rn)
from (
    select t.*, row_number() over(partition by id order by rand) rn
    from table_name t
) t
where rn <= 2

每个最多选择两条记录id，其中有最小的rand;你可以安排order by的条款row_number()根据您的实际排序标准。