【个人开发笔记记录】
简单例子
基于gender字段 将male 和 female 分租
const obj = [
{
name: '谢广坤',
gender: 'male'
},
{
name: '赵四',
gender: 'male'
},
{
name: '刘能',
gender: 'male'
},
{
name: '王小蒙',
gender: 'female'
}
]
type stringKey = Record<string, any[]>
const result: stringKey = {}
const apartGroup = () => {
for (let i = 0; i < obj.length; i++) {
const key = obj[i].gender
if (result[key]) {
result[key].push(obj[i])
} else {
result[key] = [obj[i]]
}
}
console.log('result', result)
return result
}
apartGroup()
output:
封装成公共函数:
const obj = [
{
name: '谢广坤',
gender: 'male'
},
{
name: '赵四',
gender: 'male'
},
{
name: '刘能',
gender: 'male'
},
{
name: '王小蒙',
gender: 'female'
}
]
const numList = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20]
/**
* 将数组根据指定的键名或函数分组。
* @param arr 要分组的数组,数组元素可以是任意类型。
* @param keyName 键名或函数,用于确定如何对数组元素进行分组。如果是字符串,则表示数组元素的属性名;如果是函数,则表示根据函数返回值进行分组。
* @returns 返回一个对象,每个键对应一个分组,键值是由具有相同键名或键值的数组元素组成的数组。
*/
const apartGroup = (arr: Array<any>, keyName: Function | string) => {
// 定义一个使用字符串作为键,值为任意类型数组的类型
type stringKey = Record<string, any[]>
const result: stringKey = {}
// 如果keyName是字符串,进行参数归一化,重新赋值key
if (typeof keyName === 'string') {
const key = keyName
keyName = (item: any) => item[key]
}
// 遍历数组,根据keyName获取每个元素的键值,然后根据键值将元素添加到相应的分组中
for (let i = 0; i < arr.length; i++) {
// 为当前元素生成一个键
const key = keyName(arr[i], i, arr)
// 检查结果对象中是否已存在该键
if (result[key]) {
// 如果键已存在,将当前元素添加到对应的数组中
result[key].push(arr[i])
} else {
// 如果键不存在,创建一个新的数组并将当前元素添加到该数组中
result[key] = [arr[i]]
}
}
console.log('result', result) // 打印分组结果
return result
}
// 使用字符串形式调用
// apartGroup(obj, 'gender')
// 使用函数形式调用
apartGroup(obj, (p: any) => p.gender)
apartGroup(numList, (n: any) => (n % 2 === 0 ? '偶数' : '奇数'))
output: