使用spark / java的mod函数

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

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

        // 创建一个包含整数的DataFrame
        Dataset<Row> df = spark.range(10);

        // 使用mod函数计算每个数除以3的余数
        df.withColumn("mod_result", functions.mod(df.col("id"), 3))
                .show();
    }
}