PostgreSQL array_sort로 2차원 배열이 정렬 안 될 때? 깔끔한 해결법과 샘플 코드 예시

2025-11-30

PostgreSQL에는 기본적으로 다차원 배열을 다루는 강력한 기능이 있지만, 때때로 배열 내 배열(2차원 배열)을 정렬하는 과정에서 예상치 못한 동작이나 제약 사항 때문에 어려움을 겪을 수 있습니다.

PostgreSQL의 array_sort(anyarray) 함수는 1차원 배열의 요소를 정렬하는 데 사용됩니다. 하지만 이 함수를 2차원 이상의 배열에 직접 적용하려고 하면 문제가 발생할 수 있습니다.

문제 array_sort 함수는 1차원 배열을 정렬하도록 설계되었기 때문에, 2차원 배열(배열 내 배열)을 입력하면 원하는 대로 '내부 배열 전체'를 기준으로 정렬하지 못하거나 오류를 발생시킬 수 있습니다.

원인
이 함수는 배열의 개별 요소를 비교하여 정렬하며, 2차원 배열에서는 내부 배열 전체가 하나의 요소가 되기 때문에, 기본 비교 연산자로는 어떤 기준으로 정렬해야 할지 명확하지 않습니다. (예
내부 배열의 첫 번째 요소로 정렬할지, 합계로 정렬할지 등)

문제
내부 배열을 특정 기준(예
내부 배열의 2번째 요소)으로 정렬하고 싶지만, array_sort 함수는 정렬 기준을 지정하는 매개변수를 제공하지 않습니다. (PostgreSQL의 ORDER BY 절처럼 복잡한 로직을 배열 함수 내부에서 직접 지원하지 않습니다.)

2차원 배열을 정렬해야 할 때, SQL 함수와 집계 함수를 사용하여 정렬 기준을 명시해 줄 수 있습니다. 가장 일반적인 대체 방법은 배열을 집합(SET)으로 변환하여 정렬하고, 다시 배열로 집계하는 것입니다.

이 방법은 2차원 배열을 행(Row) 형태로 풀어낸 다음, 원하는 정렬 기준을 적용하고, 마지막에 다시 배열로 묶는 (재집계) 표준적인 방식입니다.

이 예제는 내부 배열을 첫 번째 요소를 기준으로 오름차순 정렬하는 방법을 보여줍니다.

-- 2차원 배열을 받아 내부 배열의 첫 번째 요소를 기준으로 정렬하는 함수
CREATE OR REPLACE FUNCTION sort_2d_array_by_first_element(
    input_array INTEGER[][]
)
RETURNS INTEGER[][]
LANGUAGE sql
AS $$
    SELECT 
        ARRAY_AGG(sub_array ORDER BY sub_array[1] ASC)
    FROM 
        UNNEST(input_array) AS sub_array;
$$;

SELECT 
    sort_2d_array_by_first_element(
        ARRAY[[3, 10], [1, 20], [2, 30]]
    );

-- 결과 (내부 배열의 첫 번째 요소(3, 1, 2)를 기준으로 정렬됨):
-- {{1,20},{2,30},{3,10}}

UNNEST(input_array)
2차원 배열을 풀어서, 각 내부 배열([3, 10], [1, 20], [2, 30])을 별도의 행(sub_array)으로 만듭니다.

ORDER BY sub_array[1] ASC
각 행(내부 배열)을 첫 번째 요소(sub_array[1])를 기준으로 정렬합니다. 만약 두 번째 요소(sub_array[2])로 정렬하고 싶다면 이것만 변경하면 됩니다.

ARRAY_AGG(...)
정렬된 행들을 다시 하나의 2차원 배열로 집계하여 반환합니다.

함수를 생성하지 않고, 쿼리 내에서 바로 정렬을 수행하고 싶을 때 LATERAL JOIN을 사용할 수 있습니다.

WITH data_array AS (
    SELECT 
        ARRAY[[9, 100], [4, 50], [7, 80]] AS my_2d_array
)
SELECT 
    T.sorted_array
FROM 
    data_array,
    LATERAL (
        SELECT 
            ARRAY_AGG(sub_array ORDER BY sub_array[2] DESC) AS sorted_array -- 두 번째 요소를 기준으로 내림차순 정렬
        FROM 
            UNNEST(data_array.my_2d_array) AS sub_array
    ) AS T;

-- 결과 (내부 배열의 두 번째 요소(100, 50, 80)를 기준으로 내림차순 정렬됨):
-- {{9,100},{7,80},{4,50}}