如何在spark scala中读取csv文件并为变量赋值

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

// 如果尚未安装CSV文件读取库，可以使用以下命令进行安装
// spark-shell --packages com.databricks:spark-csv_2.11:1.5.0

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

val csvPath = "path/to/csv/file.csv" // 替换为实际的CSV文件路径
val df = spark.read.format("csv")
  .option("header", "true") // 如果CSV文件包含标题行，则设置为true
  .option("inferSchema", "true") // 自动推断列的数据类型
  .load(csvPath)

val variable = df.select("column_name").first().getString(0)
// 替换"column_name"为实际的列名，使用first()获取第一行数据，getString(0)获取第一列的字符串值

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

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

val csvPath = "path/to/csv/file.csv"
val df = spark.read.format("csv")
  .option("header", "true")
  .option("inferSchema", "true")
  .load(csvPath)

val variable = df.select("column_name").first().getString(0)