sumBy - 根据函数映射每个元素,然后返回数组的和

优质
小牛编辑
122浏览
2023-12-01

使用提供的函数将每个元素映射到一个值之后,然后返回数组的和。

使用 Array.map() 将每个元素映射到由 fn 返回的值,Array.reduce() 将每个值添加到一个累加器,并且累加器初始值为 0

const sumBy = (arr, fn) =>
  arr.map(typeof fn === 'function' ? fn : val => val[fn]).reduce((acc, val) => acc + val, 0);
sumBy([{ n: 4 }, { n: 2 }, { n: 8 }, { n: 6 }], o => o.n); // 20
sumBy([{ n: 4 }, { n: 2 }, { n: 8 }, { n: 6 }], 'n'); // 20