Spark dataset获取与整型列标题相同的数据

import org.apache.spark.sql.SparkSession

val spark = SparkSession.builder()
  .appName("Spark Dataset Example")
  .master("local")
  .getOrCreate()

val data = Seq(
  (1, "John", 25),
  (2, "Jane", 30),
  (3, "Bob", 35)
)

val df = spark.createDataFrame(data).toDF("id", "name", "age")

import org.apache.spark.sql.functions.col

val columnName = "age"
val filteredData = df.filter(col(columnName).isNotNull)

filteredData.show()