Làm thế nào để áp dụng một hàm cho mỗi phần tử của một cột mảng trong Postgres?

Question

Truy vấn Pg trả về một mảng. Tôi muốn truy xuất điều đó với mỗi phần tử được định dạng thành 3 chữ số thập phân. Làm thế nào tôi có thể áp dụng một hàm cho mỗi phần tử của một mảng? Một cái gì đó như sau (sai, rõ ràng) -

SELECT Round(ARRAY[1.53224,0.23411234], 2); 
{1.532, 0.234} 

Tôi đoán tôi đang tìm kiếm một cái gì đó như Perl's map chức năng.

Answer 1

Bạn có thể cần phải tạo một hàm được lưu trữ. Đây là một trong đó làm những gì bạn cần:

CREATE OR REPLACE FUNCTION array_round(float[], int) 
RETURNS float[] 
AS 
$$ 
DECLARE 
    arrFloats ALIAS FOR $1; 
    roundParam ALIAS FOR $2; 
    retVal float[]; 
BEGIN 
    FOR I IN array_lower(arrFloats, 1)..array_upper(arrFloats, 1) LOOP 
    retVal[I] := round(CAST(arrFloats[I] as numeric), roundParam); 
    END LOOP; 
RETURN retVal; 
END; 
$$ 
LANGUAGE plpgsql 
    STABLE 
RETURNS NULL ON NULL INPUT; 

Sau đó gọi một cái gì đó như thế này:

# SELECT array_round(ARRAY[1.53224,0.23411234], 2); 
array_round 
------------- 
{1.53,0.23} 
(1 row) 

Answer 2

Bạn cần phải bật mảng thành một tập hàng. Ví dụ: sử dụng generate_series:

SELECT ARRAY(SELECT ROUND(ARRAY[1.53224,0.23411234])[i], 2) FROM generate_series(1,2) AS s(i));  

Tôi biết điều đó thật xấu xí. Nên có một hàm trợ giúp để tạo các ánh xạ như vậy dễ dàng hơn.

Có lẽ một cái gì đó tương tự (có nó là khủng khiếp, chậm, và giòn mã ðộng):

CREATE OR REPLACE FUNCTION map_with_arg(TEXT, ANYARRAY, TEXT) 
RETURNS ANYARRAY 
IMMUTABLE STRICT 
LANGUAGE 'plpgsql' AS 
$$ 
DECLARE 
    i INTEGER; 
    t TEXT; 
    cmd TEXT; 
BEGIN 
    FOR i IN array_lower($2, 1) .. array_upper($2, 1) LOOP 
     cmd := 'SELECT ('||quote_ident($1)||'('||quote_nullable($2[i])||', '||quote_nullable($3)||'))::TEXT'; 
     EXECUTE cmd INTO t; 
     $2[i] := t; 
    END LOOP; 
    RETURN $2; 
END; 
$$; 

select map_with_arg('repeat', array['can','to']::TEXT[], '2'); 
map_with_arg 
--------------- 
{cancan,toto} 

Cập nhật Nó xảy ra với tôi rằng chúng ta có thể sử dụng một tuyên bố động duy nhất cho toàn bộ vòng lặp. Điều này có thể giảm thiểu một số mối quan tâm về hiệu suất.

CREATE OR REPLACE FUNCTION map_with_arg(TEXT, ANYARRAY, TEXT) 
RETURNS ANYARRAY 
IMMUTABLE STRICT 
LANGUAGE 'plpgsql' AS 
$$ 
DECLARE 
    cmd TEXT; 
    rv TEXT; 
BEGIN 
    cmd := 'SELECT ARRAY(SELECT (' || quote_ident($1)||'($1[i], '||quote_nullable($3)||'))::TEXT FROM generate_subscripts($1, 1) AS gs(i))'; 
    EXECUTE cmd USING $2 INTO rv; 
    RETURN rv; 
END; 
$$; 

Answer 3

Đầu tiên, lần lượt các mảng thành một tập sử dụng unnest:

> SELECT n FROM unnest(ARRAY[1.53224,0.23411234]) AS n; 
    n  
------------ 
    1.53224 
0.23411234 
(2 rows) 

Sau đó, áp dụng một biểu thức để cột:

> SELECT ROUND(n, 2) FROM unnest(ARRAY[1.53224,0.23411234]) AS n; 
round 
------- 
    1.53 
    0.23 
(2 rows) 

Cuối cùng, sử dụng array_agg để biến thiết lập trở lại thành một mảng:

> SELECT array_agg(ROUND(n, 2)) FROM unnest(ARRAY[1.53224,0.23411234]) AS n; 
    array_agg 
------------- 
{1.53,0.23} 
(1 row) 

Answer 4

postgres=# select array(select round(unnest(array[1.2,2.4,3,4]))); 
    array 
----------- 
{1,2,3,4} 
(1 row)