pyspark mllib IndexedRowMatrix型インデックスを使って項目を取得する方法

Question

mllib indexedRowMatrix型から行列の内容を取得することは可能ですか？例えば。私は与える(0,1)座標場合

rows = sc.parallelize([IndexedRow(0, [1, 2, 3]),IndexedRow(1, [4, 5, 6]),IndexedRow(2, [7, 8, 9]),IndexedRow(3, [10, 11, 12])]) 
mat = IndexedRowMatrix(rows) 

、私はランダムアクセスをサポートしていません（一般的には、分散コンテナ）mat

Answer 1

スパーク行列から2を取得する必要があります。

mat.rows.filter(lambda row: row.index == 0).first().vector[1] 

が、この操作は、行の数という点で線形である：それはfilterすることが可能です。