在Java Spark中使用withcolumn遍历不同的列

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

public class SparkWithColumnExample {
    public static void main(String[] args) {
        SparkSession spark = SparkSession.builder()
                .appName("SparkWithColumnExample")
                .master("local")
                .getOrCreate();
        
        // 读取CSV文件创建DataFrame
        Dataset<Row> df = spark.read()
                .option("header", "true")
                .csv("path/to/input.csv");
        
        // 使用withColumn遍历不同列
        Dataset<Row> newDF = df.withColumn("newColumn", col("columnName").plus(col("otherColumnName")));
        
        // 显示新的DataFrame
        newDF.show();
        
        spark.stop();
    }
}