sql - 枚举 Postgres 表中的表分区

Question

假设我有一个这样的表:

id  | part  | value
----+-------+-------
 1  | 0     | 8
 2  | 0     | 3
 3  | 0     | 4
 4  | 1     | 6
 5  | 0     | 13
 6  | 0     | 4
 7  | 1     | 2
 8  | 0     | 11
 9  | 0     | 15
 10 | 0     | 3
 11 | 0     | 2

我想枚举部分属性为 0 的组。

最终我想得到这个:

id  | part  | value | number
----+-------+-----------------
 1  | 0     | 8     |   1
 2  | 0     | 3     |   2
 3  | 0     | 4     |   3
 4  | 1     | 6     |   0
 5  | 0     | 13    |   1
 6  | 0     | 4     |   2
 7  | 1     | 2     |   0
 8  | 0     | 11    |   1
 9  | 0     | 15    |   2
 10 | 0     | 3     |   3
 11 | 0     | 2     |   4

是否可以使用 Postgres 窗口函数来解决这个问题，或者有其他方法吗？

Answer 1

是的，这很简单:

SELECT id, part, value,
       row_number() OVER (PARTITION BY grp ORDER BY id) - 1 AS number
FROM (SELECT id, part, value,
             sum(part) OVER (ORDER BY id) AS grp
      FROM mytable
     ) AS q;

 id | part | value | number 
----+------+-------+--------
  1 |    0 |     8 |      0
  2 |    0 |     3 |      1
  3 |    0 |     4 |      2
  4 |    1 |     6 |      0
  5 |    0 |    13 |      1
  6 |    0 |     4 |      2
  7 |    1 |     2 |      0
  8 |    0 |    11 |      1
  9 |    0 |    15 |      2
 10 |    0 |     3 |      3
 11 |    0 |     2 |      4
(11 rows)