Use 'i' for index, and zipWithIndex

ThalesGroup · Jan 21, 2015 · bbc907a · bbc907a
1 parent cb10ae5
commit bbc907a
Showing 1 changed file with 6 additions and 2 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,9 +84,13 @@ class IndexedRowMatrix(
       val rowIndex = row.index
       row.vector match {
         case SparseVector(size, indices, values) =>
-          Array.tabulate(indices.size)(x => MatrixEntry(rowIndex, indices(x), values(x)))
+          indices.zip(values).map { case (i, iVal) =>
+            MatrixEntry(rowIndex, i, iVal)
+          }
         case DenseVector(values) =>
-          Array.tabulate(values.size)(x => MatrixEntry(rowIndex, x, values(x)))
+          values.zipWithIndex.map { case (iVal, i) =>
+            MatrixEntry(rowIndex, i, iVal)
+          }
       }
     }
     new CoordinateMatrix(entries, numRows(), numCols())