scala - 如何从DataFrame中获取Array[Seq[String]]？

Question

如何将DataFrame cc传递到Array[Seq[String]]？

val factors = $(ccCols).split(",")
val cc = dataset.select(factors.head, factors.tail: _*)

我尝试了这种方式，但它给了我Array[Row]:

cc.rdd.collect()

Answer 1

您将需要使用 Row 对象的 toSeq 函数:

val a = sc.parallelize(Seq((1,2),(3,4))).toDF("a", "b")
a.show
/*
Output:
+-+-+
|a|b|
+-+-+
|1|2|
|3|4|
+-+-+
*/

a.collect.map(_.toSeq)
// Output: Array(WrappedArray(1, 2), WrappedArray(3, 4))