feat(countBy): Add countBy function

benchmarks/countBy.bench.ts

@@ -0,0 +1,17 @@
+import { bench, describe } from 'vitest';
+import { countBy as countByToolkit } from 'es-toolkit';
+import { countBy as countByLodash } from 'lodash';
+
+describe('countBy', () => {
+  bench('es-toolkit/countBy', () => {
+    countByToolkit([1.2, 2.4, 3.6, 2.2, 3.4, 3.6], (item: number) => {
+      return Math.floor(item).toString();
+    });
+  });
+
+  bench('lodash/countBy', () => {
+    countByLodash([1.2, 2.4, 3.6, 2.2, 3.4, 3.6], (item: number) => {
+      return Math.floor(item).toString()
+    });
+  });
+});

docs/ko/reference/array/countBy.md

@@ -0,0 +1,30 @@
+# countBy
+
+배열의 각 요소에에 `transformer`를 적용한 요소와 그 요소의 갯수를 key-value의 형태로 반환해요.
+
+## 인터페이스
+
+```typescript
+function countBy<T>(arr: T[], transformer: (item: T) => string): Record<string, number>
+```
+
+### 파라미터
+
+- `arr` (`T[]`): 각 요소의 갯수를 세고자 하는 배열
+- `transformer` (`(item: T) => string`): 각 요소를 key로 변환하는 함수
+
+### 반환 값
+
+(`Record<string, number>`) 각 요소와 그 갯수를 key-value로 가지는 객체
+
+## 예시
+
+```javascript
+import {countBy} from 'es-toolkit/array';
+
+const array = [1, 2, 3, 4, 5, 6];
+const result = countBy(array, x => x % 2 === 0 ? 'even' : 'odd');
+
+console.log(result);
+// 출력: { 'odd': 3, 'even': 3 }
+```

docs/reference/array/countBy.md

@@ -0,0 +1,31 @@
+# countBy
+
+Count the occurrences of each item in an array based on a transformation function.
+
+## Signature
+
+```typescript
+function countBy<T>(arr: T[], transformer: (item: T) => string): Record<string, number>
+```
+
+### Parameters
+
+- `arr` (`T[]`): The input array to count occurrences.
+- `transformer` (`(item: T) => string`): The transformation function that maps each item to a string key.
+
+### Returns
+
+(`Record<string, number>`) An object containing the count of each item based on the transformation function.
+
+## Examples
+
+```javascript
+import {countBy} from 'es-toolkit/array';
+
+const array = [1, 2, 3, 4, 5, 6];
+const result = countBy(array, x => x % 2 === 0 ? 'even' : 'odd');
+
+console.log(result);
+// Output: { 'odd': 3, 'even': 3 }
+```
+

src/array/countBy.spec.ts

@@ -0,0 +1,26 @@
+import { describe, expect, it } from 'vitest';
+import {countBy} from "./countBy.ts";
+
+describe('countBy', () => {
+  it('should count the occurrences of each item in an array', () => {
+    const arr = [1, 2, 3, 4, 5, 1, 2, 3, 4, 5];
+    const result = countBy(arr, String);
+
+    expect(result).toEqual({
+      '1': 2,
+      '2': 2,
+      '3': 2,
+      '4': 2,
+      '5': 2,
+    });
+  });
+  it('should count the occurrences of each item in an array that applied transformer', () => {
+    const arr = [1, 2, 3, 4, 5, 1, 2, 3, 4, 5];
+    const result = countBy(arr, (item) => item % 2 === 0 ? 'even' : 'odd');
+
+    expect(result).toEqual({
+      'odd': 6,
+      'even': 4,
+    });
+  })
+});

src/array/countBy.ts

@@ -0,0 +1,24 @@
+/**
+ * Count the occurrences of each item in an array
+ * based on a transformation function.
+ *
+ * This function takes an array and a transformation function
+ * that converts each item in the array to a string. It then
+ * counts the occurrences of each transformed item and returns
+ * an object with the transformed items as keys and the counts
+ * as values.
+ *
+ * @template T - The type of the items in the input array.
+ *
+ * @param {T[]} arr - The input array to count occurrences.
+ * @param {(item: T) => string} transformer - The transformation function that maps each item to a string key.
+ * @returns {Record<string, number>} An object containing the transformed items as keys and the
+ * counts as values.
+ */
+export function countBy<T>(arr: T[], transformer: (item: T) => string): Record<string, number> {
+  return arr.reduce((acc, item) => {
+    const key = transformer(item);
+    acc[key] = (acc[key] || 0) + 1;
+    return acc;
+  }, {} as Record<string, number>);
+}

src/array/index.ts

@@ -1,5 +1,6 @@
 export { chunk } from './chunk.ts';
 export { compact } from './compact.ts';
+export { countBy } from './countBy.ts';
 export { difference } from './difference.ts';
 export { differenceBy } from './differenceBy.ts';
 export { differenceWith } from './differenceWith.ts';