在Spark scala中使用列创建格式化csv文件

import org.apache.spark.sql.{SparkSession, DataFrame}
import org.apache.spark.sql.functions._

val spark = SparkSession.builder()
  .appName("CSV Writer")
  .getOrCreate()

val data = Seq(
  ("John", 25, "USA"),
  ("Alice", 30, "Canada"),
  ("Bob", 35, "UK")
)

val df = spark.createDataFrame(data).toDF("Name", "Age", "Country")

val selectedColumns = df.select("Name", "Country")

val coalescedDF = selectedColumns.coalesce(1)

coalescedDF.write
  .format("csv")
  .option("header", "true")
  .mode("overwrite")
  .save("path/to/output.csv")