Spark-Java :如何在spark Dataframe中添加数组列

import org.apache.spark.sql.Column;
import org.apache.spark.sql.functions;

Column hobbiesColumn = functions.array(functions.lit("reading"), functions.lit("gaming"), functions.lit("cooking"));
df = df.withColumn("hobbies", hobbiesColumn);

df.show();

import org.apache.spark.sql.Column;
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 SparkJavaExample {
    public static void main(String[] args) {
        SparkSession spark = SparkSession.builder()
                .appName("Spark Java Example")
                .master("local")
                .getOrCreate();

        // 创建一个示例DataFrame
        Dataset<Row> df = spark.createDataFrame(
                spark.sparkContext().parallelize(
                        java.util.Arrays.asList(
                                new java.util.ArrayList<>(java.util.Arrays.asList("John", 25)),
                                new java.util.ArrayList<>(java.util.Arrays.asList("Jane", 30)),
                                new java.util.ArrayList<>(java.util.Arrays.asList("Bob", 35))
                        )
                ),
                new org.apache.spark.sql.types.StructType()
                        .add("name", "string")
                        .add("age", "integer")
        );

        // 添加一个数组列
        Column hobbiesColumn = functions.array(functions.lit("reading"), functions.lit("gaming"), functions.lit("cooking"));
        df = df.withColumn("hobbies", hobbiesColumn);

        // 显示结果
        df.show();
    }
}