【对象分组操作】对象基于某个字段进行分组

【个人开发笔记记录】

简单例子

基于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：