具有限制的 Postgres 查詢選擇具有相似標識符的所有記錄

Question

我有一個看起來像這樣的表：

客戶ID	數據
1	123
1	456
2	789
2	101
2	121
2	123
3	123
4	456

我想做的是執行SELECT結合LIMIT X以獲得 X 條記錄以及具有相同customer_id的任何其他記錄

查詢示例： SELECT customer_id, data FROM table ORDER BY customer_id LIMIT 3; 此查詢返回：

客戶ID	數據
1	123
1	456
2	789

我想要一個查詢，該查詢將查看最后一個customer_id值並返回匹配超出指定LIMIT的所有剩余記錄。 是否有可能在一次操作中做到這一點？

所需的 output：

客戶ID	數據
1	123
1	456
2	789
2	101
2	121
2	123

Answer 1

在 Postgres 13 中可以使用with ties ：

select t.*
from t
order by customer_id
fetch first 3 rows with ties;

在早期版本in ，您可以使用：

select t.*
from t
where t.customer_id in (select t2.customer_id
                        from t t2
                        order by t2.customer_id
                        limit 3
                       );

Answer 2

您可以將相關子查詢與count一起使用，如下所示：

Select t.*
  From t
 Where 3 >= (select count(distinct customer_id)
               From t tt 
              where t.customer_id >= tt.customer_id)