Python pyspark IndexedRowMatrix.toBlockMatrix用法及代碼示例

本文簡要介紹 pyspark.mllib.linalg.distributed.IndexedRowMatrix.toBlockMatrix 的用法。

用法: toBlockMatrix(rowsPerBlock=1024, colsPerBlock=1024)

將此矩陣轉換為 BlockMatrix。

參數：

rowsPerBlock：整數，可選: 組成每個塊的行數。形成最終行的塊不需要具有給定的行數。
colsPerBlock：整數，可選: 組成每個塊的列數。形成最終列的塊不需要具有給定的列數。

例子：

>>> rows = sc.parallelize([IndexedRow(0, [1, 2, 3]),
...                        IndexedRow(6, [4, 5, 6])])
>>> mat = IndexedRowMatrix(rows).toBlockMatrix()

>>> # This IndexedRowMatrix will have 7 effective rows, due to
>>> # the highest row index being 6, and the ensuing
>>> # BlockMatrix will have 7 rows as well.
>>> print(mat.numRows())
7

>>> print(mat.numCols())
3

相關用法

注：本文由純淨天空篩選整理自spark.apache.org大神的英文原創作品 pyspark.mllib.linalg.distributed.IndexedRowMatrix.toBlockMatrix。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。