如何在Apache Spark ML API中从`DataFrame`创建一个`Vector`？

Question

如何在Apache Spark ML API中从Vector创建DataFrame（导入org.apache.spark.mllib.linalg。{DenseVector，SparseVector}）？

Answer 1

如果你想要scala.collection.Vector 您只需收集DataFrame并调用toVector()函数。

val df = Seq(1, 2, 3).toDF
val vector = df.collect().toVector
vector.map(_.getInt(0))

页。秒。请解释为什么有一个downvote