如何将数据帧拆分为固定大小的组？

import pandas as pd

# 创建一个示例数据帧
data = {
    'A': range(1, 21),
    'B': range(21, 41)
}
df = pd.DataFrame(data)

# 定义每组的大小
group_size = 5

# 计算需要多少组
num_groups = len(df) // group_size + (1 if len(df) % group_size != 0 else 0)

# 将数据帧拆分为固定大小的组
groups = [df[i*group_size:(i+1)*group_size] for i in range(num_groups)]

# 打印每个组
for i, group in enumerate(groups):
    print(f"Group {i+1}:\n{group}\n")

WITH numbered_rows AS (
    SELECT *, ROW_NUMBER() OVER (ORDER BY some_column) AS row_num
    FROM your_table
)
SELECT *
FROM numbered_rows
WHERE row_num BETWEEN start_row AND end_row;

import org.apache.spark.sql.Dataset;
import org.apache.spark.sql.Row;
import org.apache.spark.sql.SparkSession;

public class DataFrameSplitter {
    public static void main(String[] args) {
        SparkSession spark = SparkSession.builder().appName("DataFrameSplitter").getOrCreate();

        // 创建一个示例数据帧
        Dataset<Row> df = spark.range(1, 21).toDF("A");

        int groupSize = 5;
        long totalRows = df.count();
        long numGroups = (totalRows + groupSize - 1) / groupSize;

        for (int i = 0; i < numGroups; i++) {
            long startRow = i * groupSize;
            long endRow = Math.min(startRow + groupSize, totalRows);
            Dataset<Row> group = df.filter(df.col("A").between(startRow + 1, endRow));
            System.out.println("Group " + (i + 1) + ":\n" + group.show());
        }

        spark.stop();
    }
}