pyspark mllib IndexedRowMatrix類型如何通過索引獲得項目

Question

是否可以通過提供座標獲取mllib indexedRowMatrix類型的矩陣內容？例如。

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) 

如果我給座標(0,1)，我應該得到2從mat

Answer 1

星火矩陣（和分佈式容器一般）不支持隨機訪問。有可能filter：

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

但是這個操作在行數方面是線性的。