PostgreSQL中的行转列

发布时间 2023-12-11 15:42:40作者: 553490191

PG有三种行转列写法:

1、group by + sum + case when

2、用postgresql的crosstab函数

3、group by + string_agg + split_part(分组,行转列,字符切割)

环境准备

 1CREATE TABLE sales (
 2    year INTEGER,
 3    quarter INTEGER,
 4    sales_amount NUMERIC
 5);
 6
 7INSERT INTO sales VALUES
 8    (2018, 1, 100),
 9    (2018, 2, 200),
10    (2018, 3, 300),
11    (2018, 4, 400),
12    (2019, 1, 500),
13    (2019, 2, 600),
14    (2019, 3, 700),
15    (2019, 4, 800);

我们想将每个季度的销售额作为一列,年份作为行,结果:

year | q1  | q2  | q3  | q4
2------+-----+-----+-----+-----
2018 100 200 300 400
2019 500 600 700 800
5(rows)

方法1:使用crosstab函数

PostgreSQL中可以使用crosstab函数将行转列,需要使用一个额外的模块“tablefunc”,安装完成后,我们可以使用crosstab函数将行转列。

1CREATE EXTENSION tablefunc;
2
3SELECT *
4FROM crosstab(
5    'SELECT year, quarter, sales_amount
6     FROM sales
7     ORDER BY 1, 2',
8    'SELECT quarter FROM generate_series(1,4) AS quarter'
9) AS sales_pivot(year INTEGER, q1 NUMERIC, q2 NUMERIC, q3 NUMERIC, q4 NUMERIC);

在这个例子中,我们将查询作为第一个参数传递给crosstab函数。查询必须按年份和季度排序。第二个参数是一个子查询,用于生成列名。在本例中,我们使用generate_series函数生成1到4之间的数字作为季度列的名称。

crosstab函数将行转换为列,并返回一个新的表。我们在外部查询中指定每个生成的列的数据类型和名称,以便正确返回结果。

方法2:使用group by + sum + case when

使用GROUP BY + SUM + CASE WHEN 也可以实现将行转列的效果。以下是一个示例:

1SELECT
2    year,
3    SUM(CASE WHEN quarter THEN sales_amount ELSE END) AS q1,
4    SUM(CASE WHEN quarter THEN sales_amount ELSE END) AS q2,
5    SUM(CASE WHEN quarter THEN sales_amount ELSE END) AS q3,
6    SUM(CASE WHEN quarter THEN sales_amount ELSE END) AS q4
7FROM sales
8GROUP BY year
9ORDER BY year;

在这个示例中,我们使用了四个不同的CASE WHEN表达式来计算每个季度的销售额。在每个CASE WHEN表达式中,我们检查季度是否等于1、2、3或4,如果是,就将对应的销售额加入到该季度的总计中。否则,我们将0加入到总计中。

在查询中,我们使用GROUP BY子句对年份进行分组,并对每个季度的销售额进行求和。结果与使用crosstab函数得到的结果相同。

方法3:使用group by + string_agg + split_part(分组,行转列,字符切割)

使用GROUP BY + string_agg + split_part 也可以实现将行转列的效果。以下是一个示例:

 1SELECT
 2    year,
 3    split_part(sales_agg, ',', 1)::numeric AS q1,
 4    split_part(sales_agg, ',', 2)::numeric AS q2,
 5    split_part(sales_agg, ',', 3)::numeric AS q3,
 6    split_part(sales_agg, ',', 4)::numeric AS q4
 7FROM (
 8    SELECT
 9        year,
10        string_agg(sales_amount::text, ',' ORDER BY quarter) AS sales_agg
11    FROM sales
12    GROUP BY year
13) AS sales_pivot;

在这个示例中,我们使用string_agg函数将每个季度的销售额连接成一个以逗号分隔的字符串(这里一定需要加上order by子句)。然后,我们使用split_part函数将字符串拆分成四个部分,以获取每个季度的销售额,并将其转换为数字类型。最后,我们在外部查询中指定了每个季度的数据类型和名称。

在查询中,我们首先使用GROUP BY子句对年份进行分组,并使用string_agg函数将每个季度的销售额连接成一个以逗号分隔的字符串。然后,我们在外部查询中使用split_part函数将字符串拆分成四个部分,并将其转换为数字类型,以获取每个季度的销售额。结果与使用crosstab函数或GROUP BY + SUM + CASE WHEN得到的结果相同。