Spark Datasetの列名をクエリする方法は？

Question

val ds: Dataset[Double]（Spark 2.0.0）がありますが、applyまたはcolに渡してこの1列のDatasetからColumnに変換できる二重列の名前は何ですか。

Answer 1

列名はds.col("value")のように "value"です。 Dataset.schemaは、この情報が含まれています：ds.schema.fields.foreach(x => println(x))

Answer 2

ます。また、文字列の配列としてすべての列を返すDataFrameの方法columnsを、使用することができます。

case class Person(age: Int, height: Int, weight: Int){ 
    def sum = age + height + weight 
} 

val df = sc.parallelize(List(Person(1,2,3), Person(4,5,6))).toDF("age", "height", "weight") 

df.columns 
//res0: Array[String] = Array(age, height, weight)