Сгруппированный LIMIT в PostgreSQL: показать первые N строк для каждой группы?

119

Мне нужно взять первые N строк для каждой группы, упорядоченных по пользовательскому столбцу.

Учитывая следующую таблицу:

db=# SELECT * FROM xxx;
 id | section_id | name
----+------------+------
  1 |          1 | A
  2 |          1 | B
  3 |          1 | C
  4 |          1 | D
  5 |          2 | E
  6 |          2 | F
  7 |          3 | G
  8 |          2 | H
(8 rows)

Мне нужны первые 2 строки (упорядоченные по name) для каждого section_id, то есть результат, похожий на:

 id | section_id | name
----+------------+------
  1 |          1 | A
  2 |          1 | B
  5 |          2 | E
  6 |          2 | F
  7 |          3 | G
(5 rows)

Я использую PostgreSQL 8.3.5.

Теги:

5 ответов

200
Лучший ответ

Новое решение (PostgreSQL 8.4)

SELECT
  * 
FROM (
  SELECT
    ROW_NUMBER() OVER (PARTITION BY section_id ORDER BY name) AS r,
    t.*
  FROM
    xxx t) x
WHERE
  x.r <= 2;
Поделиться
  • 8
    Это работает и с PostgreSQL 8.4 (оконные функции начинаются с 8.4).
  • 2
    Слишком хорошо, чтобы использовать его пример :)
Показать ещё 3 комментария
13

Так как v9.3 вы можете выполнить боковое соединение

select distinct t_outer.section_id, t_top.id, t_top.name from t t_outer
join lateral (
    select * from t t_inner
    where t_inner.section_id = t_outer.section_id
    order by t_inner.name
    limit 2
) t_top on true
order by t_outer.section_id;

может быть быстрее, но, конечно же, вы должны проверить производительность специально на своих данных и использовать случай.

Поделиться
  • 0
    Я не очень работаю. Я получаю точно такой же вывод, когда выполняю запрос остальных пользователей, исключая часть бокового соединения, то есть беря только первую и последнюю строки вашего запроса и выполняя их. Эти два следующих возвращают мне один и тот же набор результатов: select distinct a.customer_name, a.order_date from sales a join lateral ( select * from sales b where b.customer_name = a.customer_name order by order_date desc limit 3 ) c on true order by a.customer_name; и select distinct customer_name, order_date from sales order by customer_name;
  • 0
    О, моя ошибка Это сработало, мне пришлось использовать c.order_date вместо a.order_date.
Показать ещё 3 комментария
11

Здесь другое решение (PostgreSQL <= 8.3).

SELECT
  *
FROM
  xxx a
WHERE (
  SELECT
    COUNT(*)
  FROM
    xxx
  WHERE
    section_id = a.section_id
  AND
    name <= a.name
) <= 2
Поделиться
2 
        -- ranking without WINDOW functions
-- EXPLAIN ANALYZE
WITH rnk AS (
        SELECT x1.id
        , COUNT(x2.id) AS rnk
        FROM xxx x1
        LEFT JOIN xxx x2 ON x1.section_id = x2.section_id AND x2.name <= x1.name
        GROUP BY x1.id
        )
SELECT this.*
FROM xxx this
JOIN rnk ON rnk.id = this.id
WHERE rnk.rnk <=2
ORDER BY this.section_id, rnk.rnk
        ;

        -- The same without using a CTE
-- EXPLAIN ANALYZE
SELECT this.*
FROM xxx this
JOIN ( SELECT x1.id
        , COUNT(x2.id) AS rnk
        FROM xxx x1
        LEFT JOIN xxx x2 ON x1.section_id = x2.section_id AND x2.name <= x1.name
        GROUP BY x1.id
        ) rnk
ON rnk.id = this.id
WHERE rnk.rnk <=2
ORDER BY this.section_id, rnk.rnk
        ;
Поделиться
  • 0
    CTE и оконные функции были представлены в одной и той же версии, поэтому я не вижу преимущества первого решения.
  • 0
    Посту три года. Кроме того, все еще могут быть реализации, в которых их нет (подтолкни, скажи больше). Это также можно считать упражнением в старомодном построении запросов. (хотя CTE не очень старомодны)
Показать ещё 4 комментария
2 
SELECT  x.*
FROM    (
        SELECT  section_id,
                COALESCE
                (
                (
                SELECT  xi
                FROM    xxx xi
                WHERE   xi.section_id = xo.section_id
                ORDER BY
                        name, id
                OFFSET 1 LIMIT 1
                ),
                (
                SELECT  xi
                FROM    xxx xi
                WHERE   xi.section_id = xo.section_id
                ORDER BY 
                        name DESC, id DESC
                LIMIT 1
                )
                ) AS mlast
        FROM    (
                SELECT  DISTINCT section_id
                FROM    xxx
                ) xo
        ) xoo
JOIN    xxx x
ON      x.section_id = xoo.section_id
        AND (x.name, x.id) <= ((mlast).name, (mlast).id)
Поделиться
  • 0
    Я получаю: ОШИБКА: синтаксическая ошибка в или около "JOIN"
  • 0
    @ Kouber: см. Сообщение об обновлении
Показать ещё 6 комментариев

Ещё вопросы

Сообщество Overcoder
Наверх
Меню