org.apache.spark.mllib.linalg.Vector到DataFrame scala

Question

我的输出如下

val output = Vectors.dense(Vectors.dense(pred.toDouble).toArray ++ predProb.toArray)
output: org.apache.spark.mllib.linalg.Vector = [0.0,0.9279938116118619,0.07200618838813808]

我想将其转换为DataFrame，并将以下内容作为列值

case class Prob(Pred: Double, NoProb: Double, YesProb: Double)

有人可以帮忙吗。

由于

巴拉

Answer 1

val df = output.map（_。toArray）.map {case Array（p0，p1，p2）=＆gt;问题（p0，p1，p2）}。toDF（）