如何在同一个 table 中对 GROUP 进行 LIMIT？

Question

虽然我在这里阅读了所有类似的问题，但我不知道如何在组内进行限制。阅读 PSQL 文档也没有帮助 :( 考虑以下内容：

CREATE TABLE article_relationship
(
    article_from INT NOT NULL,
    article_to INT NOT NULL,
    score INT
);

我想获得按分数排序的每个给定文章 ID 的前 5 篇相关文章的列表。

这是我尝试过的：

select DISTINCT o.article_from
from article_relationship o
join lateral (
       select i.article_from, i.article_to, i.score from article_relationship i
       order by score desc
       limit 5
       ) p on p.article_from = o.article_from
where o.article_from IN (18329382, 61913904, 66538293, 66540477, 66496909)
order by o.article_from;

它returns什么都没有。我的印象是外部查询就像循环，所以我想我只需要那里的源 ID。

此外，如果我想在 articles table 上加入 id 和 title 列并在结果集中获取相关文章的标题怎么办？

我在内部查询中添加了连接：

select o.id, p.*
from articles o
join lateral (
       select a.title, i.article_from, i.article_to, i.score
       from article_relationship i
       INNER JOIN articles a on a.id = i.article_to
       where i.article_from = o.id
       order by score desc
       limit 5
       ) p on true
where o.id IN (18329382, 61913904, 66538293, 66540477, 66496909)
order by o.id;

但这让它变得非常非常慢。

Answer 1

您的查询中没有行 return 的问题是您的连接条件错误：ON p.article_from = o.article_from；这显然应该是 ON p.article_from = o.article_to.

撇开这个问题不谈，您的查询不会 return 每个文章 ID 的前 5 个评分关系；相反，它将 return 引用整个 table 中评分最高的 5 篇引用文章之一的文章 ID，以及（还有）您指定 ID 的 5 篇引用文章中的至少一篇。

您可以使用 window 函数对每篇参考文章的评分最高的 5 篇参考文章进行排序，以对子 select 中的分数进行排名，然后 select 只有前 5 篇主要查询。有效地指定引用文章 ID 列表意味着您将对每篇引用文章的这些引用文章的评分方式进行排名：

SELECT article_from, article_to, score
FROM (
    SELECT article_from, article_to, score, 
           rank() OVER (PARTITION BY article_from ORDER BY score DESC) AS rnk
    FROM article_relationship
    WHERE article_to IN (18329382, 61913904, 66538293, 66540477, 66496909) ) a
WHERE rnk < 6
ORDER BY article_from, score DESC;

这与您的代码不同，因为它 return 每个 article_from 最多 5 条记录，但它与您的初始描述一致。

从 table articles 添加列在主查询中很容易完成：

SELECT a.article_from, a.article_to, a.score, articles.*
FROM (
    SELECT article_from, article_to, score, 
           rank() OVER (PARTITION BY article_from ORDER BY score DESC) AS rnk
    FROM article_relationship
    WHERE article_to IN (18329382, 61913904, 66538293, 66540477, 66496909) ) a
JOIN articles ON articles.id = a.article_to
WHERE a.rnk < 6
ORDER BY a.article_from, a.score DESC;

Answer 2

版本 join lateral

select o.id as from_id, p.article_to as to_id, a.title, a.journal_id, a.pub_date_p from articles o
  join lateral (
       select i.article_to from article_relationship i
       where i.article_from = o.id
       order by score desc
       limit 5
       ) p on true
  INNER JOIN articles a on a.id = p.article_to
where o.id IN (18329382, 61913904, 66538293, 66540477, 66496909)
order by o.id;

如何在同一个 table 中对 GROUP 进行 LIMIT？

How do I do LIMIT within GROUP in the same table?

postgresql

group-by

lateral