Files
30-seconds-of-code/snippets/zipWith.md
Rohit Tanwar 994e14cd0f zipWith
2018-01-20 20:51:34 +05:30

807 B

zipWith

This method is like zip except that it accepts a function (fn) as the last value to specify how grouped values should be combined.

The function is invoked with the elements of each group: (...group).

const zipWith = (...arrays) => {
  const length = arrays.length;
  let fn = length > 1 ? arrays[length - 1] : undefined;
  fn = typeof fn == 'function' ? (arrays.pop(), fn) : undefined;
  const maxLength = Math.max(...arrays.map(x => x.length));
  const result =  Array.from({ length: maxLength }).map((_, i) => {
    return Array.from({ length: arrays.length }, (_, k) => arrays[k][i]);
  })
  return fn ? result.map(arr => fn(...arr)) : result;
}
zipWith([1, 2], [10, 20], [100, 200], (a,b,c) => a + b + c); // [111,222]