This function groups an array of objects by a certain key.
JavaScript:
let groupBy = (arr, key) => arr.reduce((result, obj) => ({ ...result, [obj[key]]: (result[obj[key]] || []).concat(obj) }), {});
// Example
console.log(groupBy([{ age: 20, name: 'Jack' }, { age: 20, name: 'Jane' }, { age: 21, name: 'John' }], 'age')); // { 20: [{ age: 20, name: 'Jack' }, { age: 20, name: 'Jane' }], 21: [{ age: 21, name: 'John' }] }
console.log(groupBy([{ type: 'apple', color: 'red' }, { type: 'banana', color: 'yellow' }, { type: 'apple', color: 'green' }], 'type')); // { apple: [{ type: 'apple', color: 'red' }, { type: 'apple', color: 'green' }], banana: [{ type: 'banana', color: 'yellow' }] }
TypeScript:
let groupBy = (arr: T[], key: K): { [P in T[K]]?: T[] } => arr.reduce((result, obj) => ({ ...result, [obj[key]]: (result[obj[key]] || []).concat(obj) }), {});
// Example
console.log(groupBy([{ age: 20, name: 'Jack' }, { age: 20, name: 'Jane' }, { age: 21, name: 'John' }], 'age')); // { 20: [{ age: 20, name: 'Jack' }, { age: 20, name: 'Jane' }], 21: [{ age: 21, name: 'John' }] }
console.log(groupBy([{ type: 'apple', color: 'red' }, { type: 'banana', color: 'yellow' }, { type: 'apple', color: 'green' }], 'type')); // { apple: [{ type: 'apple', color: 'red' }, { type: 'apple', color: 'green' }], banana: [{ type: 'banana', color: 'yellow' }] }