averageBy - 根据函数映射每个元素,然后求数组的平均值

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

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

使用 Array.map() 将每个元素映射到由 fn 返回的值,Array.reduce() 将每个值累加到累加器,用 0 作为累加器的初始值,再除以数组的 length

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