2020-03-24
描述
数组中的每一个元素根据所提供的函数可获取每一项的值,然后求这些值的平均数。
提示
- 使用
Array.prototype.map()
迭代每一个元素通过fn
计算后的返回值 - 使用
Array.prototype.reduce()
对每一个返回值进行累加,初始值为 0 - 将累加值除以数组的
length
代码
const averageBy = (arr, fn) =>
arr.map(typeof fn === 'function' ? fn : val => val[fn]).reduce((acc, val) => acc + val, 0) /
arr.length;
示例
计算数组对象中 n 的平均值:
averageBy([{ n: 4 }, { n: 2 }, { n: 8 }, { n: 6 }], o => o.n); // 5
averageBy([{ n: 4 }, { n: 2 }, { n: 8 }, { n: 6 }], 'n'); // 5