如何无需逐个指定列，查询Python编写的SQL函数返回的所有列？

要解决不固定列数的查询问题，核心是使用动态SQL——静态SQL无法处理未知数量的列，而PostgreSQL的EXECUTE语句可以帮你动态生成查询逻辑。

解决方案1：用DO块自动执行动态查询 #

这个方法会自动获取列数、生成对应列的查询语句并执行，无需手动拼接：

DO $$
DECLARE
    col_count INT := test_00.test_func('shape')::INT;
    col_list TEXT;
BEGIN
    -- 生成所有列的表达式（如elem->>0 as column1, elem->>1 as column2...）
    SELECT string_agg(
        format('cast(elem->>%s as text) as column%s', i, i+1),
        ', '
    ) INTO col_list
    FROM generate_series(0, col_count - 1) AS i;

    -- 构造完整查询，用CTE缓存函数结果避免重复调用
    EXECUTE format(
        'WITH data AS (
            SELECT jsonb_array_elements(test_00.test_func(''array'')::jsonb) AS elem
        )
        SELECT %s FROM data',
        col_list
    );
END $$;

解决方案2：生成动态SQL语句后手动执行 #

如果你需要先预览生成的SQL再执行，可以用这个方法：

-- 生成动态查询语句
SELECT format(
    'WITH data AS (
        SELECT jsonb_array_elements(test_00.test_func(''array'')::jsonb) AS elem
    )
    SELECT %s FROM data',
    string_agg(
        format('cast(elem->>%s as text) as column%s', i, i+1),
        ', '
    )
) AS dynamic_query
FROM generate_series(0, (test_00.test_func('shape')::INT) - 1) AS i;

运行后会得到完整的SQL语句，复制出来直接执行即可得到拆分后的多列结果。

关键说明 #

1. 避免重复调用函数：用CTE（WITH data AS (...)）缓存test_00.test_func('array')的结果，避免多次调用函数带来的性能损耗。
2. 动态列生成逻辑：通过generate_series生成0到列数-1的索引，再用string_agg拼接成所有列的查询表达式，替代手动逐个编写列名。
3. 原代码问题解析：你之前的generate_series写法是把每个列的内容作为单独的行返回，而动态SQL实现了“行转列”的动态需求，将结果拆分为独立列。

内容的提问来源于stack exchange，提问作者Sterioz