Array -> Iterator

YanTangZhai · Jan 21, 2015 · 3ce0b5d · 3ce0b5d
1 parent bbc907a
commit 3ce0b5d
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/linalg/distributed/IndexedRowMatrix.scala b/mllib/src/main/scala/org/apache/spark/mllib/linalg/distributed/IndexedRowMatrix.scala
@@ -84,13 +84,9 @@ class IndexedRowMatrix(
       val rowIndex = row.index
       row.vector match {
         case SparseVector(size, indices, values) =>
-          indices.zip(values).map { case (i, iVal) =>
-            MatrixEntry(rowIndex, i, iVal)
-          }
+          Iterator.tabulate(indices.size)(x => MatrixEntry(rowIndex, indices(x), values(x)))
         case DenseVector(values) =>
-          values.zipWithIndex.map { case (iVal, i) =>
-            MatrixEntry(rowIndex, i, iVal)
-          }
+          Iterator.tabulate(values.size)(x => MatrixEntry(rowIndex, x, values(x)))
       }
     }
     new CoordinateMatrix(entries, numRows(), numCols())