如何在Spark中根据列值生成对应行数？

嘿，这个需求其实挺常见的，我给你几种不同数据库环境下的实现方案，你可以根据自己用的数据库来选择：

方法1：递归CTE（通用方案，适配PostgreSQL、MySQL 8+、SQL Server等主流数据库） #

递归公共表表达式（CTE）是最通用的实现方式，不需要依赖额外的辅助表就能搞定：

WITH RECURSIVE num_seq AS (
    SELECT 1 AS id
    UNION ALL
    SELECT id + 1 FROM num_seq WHERE id < (SELECT MAX(id) FROM your_table)
)
SELECT id FROM num_seq;

简单说下逻辑：先初始化起始值为1，然后递归地每次给当前id加1，直到数值达到原表中id的最大值（这里原表id是4，所以会生成1到4的连续行）。

方法2：用数据库内置序列生成函数（PostgreSQL专属） #

如果用的是PostgreSQL，那有更简洁的方式，直接用generate_series函数生成指定范围的序列：

SELECT generate_series(1, (SELECT id FROM your_table)) AS id;

这个一步到位，直接指定起始值1，结束值取原表的id值就行。

方法3：数字辅助表（适配所有数据库，适合频繁使用的场景） #

要是你的数据库不支持递归或者序列函数，那可以先建一个包含连续数字的辅助表，之后就能反复用了：
首先创建并填充数字表（这是一次性操作）：

CREATE TABLE numbers (n INT PRIMARY KEY);
-- PostgreSQL可以直接用这个插入数据
INSERT INTO numbers SELECT * FROM generate_series(1, 1000);
-- 要是用MySQL，可以用循环或者CTE来插入连续数字；SQL Server也可以用递归CTE生成

然后查询的时候就简单了：

SELECT n AS id FROM numbers WHERE n <= (SELECT id FROM your_table);

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