如何将单列IndexedRowMatrix转换为数组?

时间:2018-06-07 11:46:25

标签: scala apache-spark

转换单列分布式矩阵(在本例中为IndexedRowMatrix)的有效方法是什么,因为矩阵的索引对应于数组的索引?

我当前的方法在执行后不会更改数组,即数组保留它初始化的值(可以理解,矩阵和数组的元素数量相等):

var array = Array.fill(10)(0.0)
val matrix = new IndexedRowMatrix(...)
matrix.rows.foreach(row => array(row.index.toInt) = row.vector(0))

请注意,我不坚持使用Array,Vector类型也可以接受。

我更喜欢在没有可变数组变量的情况下另外管理的答案,但这是“奖励积分”附录。

0 个答案:

没有答案