在 SQL 查询中采样

Question

我正在解决一个类似这样的问题：

我有一个包含许多列的表，但主要是 DepartmentId 和 EmployeeIds

Employee Ids    Department Ids
------------------------------
A                   1
B                   1
C                   1
D                   1
AA                  2
BB                  2
CC                  2
A1                  3
B1                  3
C1                  3
D1                  3

我想编写一个 SQL 查询，以便为每个 DepartmentID 取出 2 个样本 EmployeeIds。

喜欢

Employee Id  Dept Ids
B              1
C              1
AA             2
CC             2
D1             3
A1             3

目前我正在编写查询，

select
   EmployeeId, DeptIds, count(*)
from 
   table_name
group by 1,2
sample 2

但它总共给了我两行。

有什么帮助吗？

Answer 1

如果我知道的部门数量少，您可以进行分层抽样：

select *
from table_name
sample
   when DeptIds = 1 then 2
   when DeptIds = 2 then 2
   when DeptIds = 3 then 2
end

否则是 RANDOM 和 ROW_NUMBER 的组合：

select *
from
 (
   sel EmployeeId, DeptIds, random(1,10000000) as rand
   from table_name
 ) as dt
qualify
   row_number()
   over (partition by DeptIds
         order by rand) <= 2