python - Pyspark 将 RowMatrix 转换为 DataFrame 或 RDD

Question

我有一个方形的 pyspark RowMatrix，如下所示:

>>> row_mat.numRows()
100
>>> row_mat.numCols()
100
>>> row_mat.rows.first()
SparseVector(100, {0: 0.0, 1: 0.0018, 2: 0.1562, 3: 0.0342...})

我想运行 pyspark.ml.feature.PCA , 但它的 fit()方法只接受一个 DataFrame。有没有办法将此 RowMatrix 转换为 DataFrame？

或者有更好的方法吗？

Answer 1

使用:

row_mat.rows.map(lambda x: (x, )).toDF()