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

Question

I have an output like the following

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

I want to convert that into a DataFrame with the following as column values

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

Can somebody please help.

Thanks

Bala

Answer 1

val df = output.map（_。toArray）.map {case Array（p0，p1，p2）=> Prob（p0，p1，p2）}。toDF（）