PostgreSQL – NTILE函数

在 PostgreSQL 中，NTILE 函数用于将结果集中的行分成指定数量的桶，并为每个桶分配一个排名值，以使每个桶中的行的数量相同。

语法

NTILE(n) OVER (
    [PARTITION BY partition_expression, ... ]
    ORDER BY sort_expression [ASC|DESC], ...
)

参数：

• n: 桶的数量。
• partition_expression: 可选，用于分组的表达式。
• sort_expression: 用于排序的表达式，将行按照这些表达式进行排序。

示例

假设有一张员工表，我们想要将所有员工分成三个桶，每个桶中的员工数量尽量相同，在每个桶中按照工资从高到低排序。

CREATE TABLE employees (
     id INTEGER,
     name VARCHAR(20),
     salary NUMERIC
);

INSERT INTO employees VALUES
(1, 'Alice', 50000),
(2, 'Bob', 55000),
(3, 'Charlie', 62000),
(4, 'David', 63000),
(5, 'Eva', 70000),
(6, 'Frank', 71000),
(7, 'Grace', 72000),
(8, 'Henry', 76000);

使用 NTILE 函数查询员工表，并将结果分成三个桶：

SELECT id, name, salary, NTILE(3) OVER (
    ORDER BY salary DESC
) AS bucket
FROM employees;

结果如下：

 id |  name   | salary | bucket 
----+---------+--------+--------
  8 | Henry   |  76000 |      1
  7 | Grace   |  72000 |      1
  6 | Frank   |  71000 |      2
  5 | Eva     |  70000 |      2
  4 | David   |  63000 |      2
  3 | Charlie |  62000 |      3
  2 | Bob     |  55000 |      3
  1 | Alice   |  50000 |      3

可以看到，结果被分成了三个桶，每个桶中的员工数量尽量相同，并且在每个桶中按照工资从高到低排序。

总结

NTILE 函数是 PostgreSQL 中非常有用的窗口函数之一，它可以将结果集中的行分成指定数量的桶，并为每个桶分配一个排名值，以使每个桶中的行的数量相同。在处理数据时，NTILE 函数可以帮助我们更好地理解数据的分布情况，从而更好地进行其他数据操作。