不拘一格解读Lodash源码精髓--countBy函数剖析

2024-01-12 04:42:40

揭秘 Lodash 中的 countBy 函数：统计元素出现次数的强大工具

简介

在 JavaScript 开发中，处理数据时经常需要统计特定元素或属性值出现的次数。lodash 中的 countBy 函数就是一项极有价值的工具，可以轻松完成这项任务。本文将深入探讨 countBy 函数的源代码，揭示其内部运作机制，并提供实际应用示例。

源码剖析

countBy 函数源码如下：

function countBy(collection, iteratee) {
  var result = {};
  if (collection == null) {
    return result;
  }
  iteratee = typeof iteratee == 'function' ? iteratee : identity;
  each(collection, function(value, key, collection) {
    var key = iteratee(value, key, collection);
    result[key] = (result[key] || 0) + 1;
  });
  return result;
}

函数结构

countBy 函数接受两个参数：

collection: 要统计的集合，可以是数组、对象或字符串。
iteratee: 一个函数，用于将集合中的元素转换为统计的键值。如果省略此参数，则使用恒等函数，即返回元素本身。

算法流程

countBy 函数内部流程如下：

创建一个空对象 result 来存储统计结果。
遍历 collection 中的每个元素。
使用 iteratee 函数将元素转换为键值。
如果 result 中已存在该键值，则将其值加 1。
否则，将键值添加到 result 中，并将其值设置为 1。
最终返回 result 对象。

应用场景

countBy 函数在实际开发中应用广泛，包括：

统计数组中元素出现的次数。
统计对象中属性值出现的次数。
统计字符串中字符出现的次数。
根据特定条件对集合中的元素进行分组和统计。

示例

让我们通过一个示例来了解如何使用 countBy 函数：

const numbers = [1, 2, 3, 4, 5, 1, 2, 3];
const result = countBy(numbers);
console.log(result); // { '1': 2, '2': 2, '3': 2, '4': 1, '5': 1 }

其他示例

以下是其他一些应用 countBy 函数的示例：

// 统计对象中属性值出现的次数
const users = [{ age: 20 }, { age: 30 }, { age: 20 }, { age: 40 }];
const result = countBy(users, (user) => user.age);
console.log(result); // { '20': 2, '30': 1, '40': 1 }

// 统计字符串中字符出现的次数
const str = 'Hello World';
const result = countBy(str, (char) => char);
console.log(result); // { 'H': 1, 'e': 1, 'l': 3, 'o': 2, 'W': 1, 'r': 1, 'd': 1 }