如何在Scala/Spark中将带有字符串的列转换为Array[String]？

import org.apache.spark.sql.functions._

// 假设有一个DataFrame df，包含一个名为str的字符串列
val df = spark.createDataFrame(Seq(
  (1, "apple,banana,orange"),
  (2, "cat,dog,rabbit"),
  (3, "red,green,blue")
)).toDF("id", "str")

// 使用split函数将字符串列转换为Array[String]
val result = df.withColumn("array", split(col("str"), ","))

// 打印结果
result.show(false)

+---+-----------------+-------------------+
|id |str              |array              |
+---+-----------------+-------------------+
|1  |apple,banana,orange|[apple, banana, orange]|
|2  |cat,dog,rabbit   |[cat, dog, rabbit]  |
|3  |red,green,blue   |[red, green, blue]  |
+---+-----------------+-------------------+