如何从2d数组中选择1d数组?

时间:2014-09-23 13:21:33

标签: arrays postgresql

我有2d数组,只想选择它的第一个元素,即1d数组 我该怎么做?

1 个答案:

答案 0 :(得分:10)

获取array的第一个切片

SELECT my_arr[1:1];

生成的数组具有相同的数组维度作为输入 我之前的答案详情如下:

展平结果:

SELECT ARRAY(SELECT unnest(my_arr[1:1]));

或清洁:

SELECT ARRAY(SELECT * FROM unnest(my_arr)[1:1]));

实施例

SELECT (ARRAY[[1,2,3], [4,5,6], [7,8,9]])[1:1];

结果:

{{1,2,3}}  -- 2D array

或者:

SELECT ARRAY(
   SELECT unnest((ARRAY[[1,2,3], [4,5,6], [7,8,9]])[1:1])
   );

结果:

{1,2,3}  -- 1D array

在Postgres 8.3中模拟unnest()

回复你的评论:
Wiki page you are linking to有点误导。我用二维数组的代码更新了它。

对于1维数组

unnest()

CREATE OR REPLACE FUNCTION unnest_1d(anyarray)
  RETURNS SETOF anyelement AS
$func$
SELECT $1[i]
FROM   generate_series(array_lower($1,1), array_upper($1,1)) i
$func$  LANGUAGE sql IMMUTABLE;

unnest()用于二维数组:

CREATE OR REPLACE FUNCTION unnest_2d(anyarray)
  RETURNS SETOF anyelement AS
$func$
SELECT $1[d1][d2]
FROM   generate_series(array_lower($1,1), array_upper($1,1)) d1
    ,  generate_series(array_lower($1,2), array_upper($1,2)) d2
$func$  LANGUAGE sql IMMUTABLE;

默认情况下,Postgres 8.3中未安装聚合函数array_agg()

CREATE AGGREGATE array_agg(anyelement) (
 SFUNC = array_append,
 STYPE = anyarray,
 INITCOND = '{}'
);

Unnest 2d数组到1d数组:

CREATE OR REPLACE FUNCTION unnest_2d_1d(anyarray)
  RETURNS SETOF anyarray AS
$func$
SELECT array_agg($1[d1][d2])
FROM   generate_series(array_lower($1,1), array_upper($1,1)) d1
    ,  generate_series(array_lower($1,2), array_upper($1,2)) d2
GROUP  BY d1
ORDER  BY d1
$func$  LANGUAGE sql IMMUTABLE;

SQL Fiddle.