Spark将数据帧转换为RowMatrix

Question

假设我有一系列转换产生的数据框。 看起来如下：

id matrixRow
0  [1,2,3]
1  [4,5,6]
2  [7,8,9]

每行实际上对应于矩阵的一行。 如何将数据框的matrixRow列转换为RowMatrix？

Answer 1

经过无数次尝试，以下是一种解决方案：

val rdd = df.rdd.map(
           row => Vectors.dense(row.getAs[Seq[Double]](1).toArray)//get the second column value as Seq[Double], then as Array, then cast to Vector
          )
val row = new RowMatrix(rdd)