30-seconds-of-code/snippets/zipWith.md

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]
zipWith([1, 2, 3], [10, 20], [100, 200], (a,b,c) => (a != null ? a : 'a') + (b != null ? b:'b') + (c != null ? c : 'c')); // [111, 222, '3bc]