在Scala中使用dataset创建CSV

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

val spark = SparkSession.builder()
  .appName("CSV Creation")
  .master("local")
  .getOrCreate()

case class Person(name: String, age: Int, city: String)

val data = Seq(
  Person("John", 25, "New York"),
  Person("Alice", 30, "London"),
  Person("Bob", 35, "Paris")
)
val dataset = spark.createDataset(data)

dataset.write
  .option("header", "true") // 包含列名
  .csv("path/to/output.csv")