Back to map

steelbreeze · Dec 8, 2024 · 21a78ea · 21a78ea
1 parent e5b4d7a
commit 21a78ea
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 11 deletions.
diff --git a/docs/dist/pivot.min.js b/docs/dist/pivot.min.js
diff --git a/lib/node/index.js b/lib/node/index.js
@@ -31,16 +31,15 @@ const criteria = (key) => (criterion) => (value) => value[key] === criterion;
 exports.criteria = criteria;
 // the implementation of pivot
 function pivot(source, ...[dimension, ...dimensions]) {
-    const matrix = [];
-    for (const criteria of dimension) {
+    const matrix = dimension.map((criteria) => {
         const slice = [];
         for (const value of source) {
             if (criteria(value)) {
                 slice.push(value);
             }
         }
-        matrix.push(slice);
-    }
+        return slice;
+    });
     // recurse if there are other dimensions, otherwise just return the matrix
     return dimensions.length ? matrix.map(slice => pivot(slice, ...dimensions)) : matrix;
 }

diff --git a/lib/web/pivot.min.js b/lib/web/pivot.min.js
diff --git a/src/index.ts b/src/index.ts
@@ -133,9 +133,7 @@ export function pivot<TValue>(source: Array<TValue>, ...dimensions: Array<Dimens
 
 // the implementation of pivot
 export function pivot<TValue>(source: Array<TValue>, ...[dimension, ...dimensions]: Array<Dimension<TValue>>) {
-	const matrix: Matrix<TValue> = [];
-
-	for (const criteria of dimension) {
+	const matrix: Matrix<TValue> = dimension.map((criteria: Criteria<TValue>) => {
 		const slice: Array<TValue> = [];
 
 		for (const value of source) {
@@ -144,8 +142,8 @@ export function pivot<TValue>(source: Array<TValue>, ...[dimension, ...dimension
 			}
 		}
 
-		matrix.push(slice);
-	}
+		return slice;
+	});
 
 	// recurse if there are other dimensions, otherwise just return the matrix
 	return dimensions.length ? matrix.map(slice => pivot(slice, ...dimensions)) : matrix;