核心答案是使用 reduce 方法将数组转换为键值对象，1. 默认后一个对象覆盖前一个处理 key 冲突；2. 可通过 merge 参数将冲突值存储为数组；3. 对于缺失 key 的情况可指定默认键存储；4. 支持通过 split 和循环访问嵌套属性作为 key。转换结果以指定属性值为键，原对象为值，完整形成键值映射对象。
将 JavaScript 对象数组转换为键值对象，核心在于利用数组的
reduce
方法，根据指定的
key
提取每个对象的值作为新对象的键，对象本身作为值。 简单来说，就是把数组里的每个东西，按照你指定的属性，变成一个对象的 key。
function keyBy(array, key) { return array.reduce((obj, item) => { obj[item[key]] = item; return obj; }, {}); } // 示例 const users = [ { id: '1', name: 'Alice' }, { id: '2', name: 'Bob' }, { id: '3', name: 'Charlie' } ]; const usersById = keyBy(users, 'id'); console.log(usersById); // 输出: // { // '1': { id: '1', name: 'Alice' }, // '2': { id: '2', name: 'Bob' }, // '3': { id: '3', name: 'Charlie' } // }
如何处理 key 值冲突的情况？
默认情况下，后面的对象会覆盖前面的对象。 如果你需要处理冲突，可以在
reduce
函数中加入判断逻辑。 比如，你可以选择保留第一个对象，或者将冲突的对象合并成一个数组。
function keyBy(array, key, merge = false) { return array.reduce((obj, item) => { const keyValue = item[key]; if (obj[keyValue]) { if (merge) { if (Array.isArray(obj[keyValue])) { obj[keyValue].push(item); } else { obj[keyValue] = [obj[keyValue], item]; } } // 如果不merge，就直接覆盖，这是默认行为 } else { obj[keyValue] = item; } return obj; }, {}); } // 示例，模拟id冲突 const users = [ { id: '1', name: 'Alice' }, { id: '2', name: 'Bob' }, { id: '1', name: 'Charlie' } ]; const usersById = keyBy(users, 'id', true); // 开启merge console.log(usersById); // 输出: // { // '1': [ { id: '1', name: 'Alice' }, { id: '1', name: 'Charlie' } ], // '2': { id: '2', name: 'Bob' } // }
如果 key 的值不存在怎么办？
如果指定的
key
在某些对象中不存在，那么对应的值将会是
undefined
。 你可以在
reduce
函数中添加一个检查，跳过这些对象，或者赋予一个默认值。
function keyBy(array, key, defaultValue = null) { return array.reduce((obj, item) => { const keyValue = item[key]; if (keyValue === undefined || keyValue === null) { // 可以选择跳过，或者赋予默认值 // return obj; // 跳过 obj[defaultValue] = item; // 赋予默认值，所有缺失key的对象都会被归到defaultValue下 } else { obj[keyValue] = item; } return obj; }, {}); } // 示例，模拟key缺失 const users = [ { id: '1', name: 'Alice' }, { name: 'Bob' }, // 缺少 id { id: '3', name: 'Charlie' } ]; const usersById = keyBy(users, 'id', 'unknown'); console.log(usersById); // 输出: // { // '1': { id: '1', name: 'Alice' }, // 'unknown': { name: 'Bob' }, // '3': { id: '3', name: 'Charlie' } // }
如何处理 key 是嵌套属性的情况？
如果 key 是嵌套的，比如
address.city
，你可以使用
lodash
的
_.get
方法，或者自己实现一个简单的函数来获取嵌套属性的值。
function keyBy(array, key) { return array.reduce((obj, item) => { // 简单实现嵌套属性的获取 const keys = key.split('.'); let value = item; for (const k of keys) { if (value && typeof value === 'object' && k in value) { value = value[k]; } else { value = undefined; break; } } if (value !== undefined) { obj[value] = item; } return obj; }, {}); } // 示例，模拟嵌套key const users = [ { id: '1', name: 'Alice', address: { city: 'New York' } }, { id: '2', name: 'Bob', address: { city: 'Los Angeles' } }, { id: '3', name: 'Charlie', address: { city: 'New York' } } ]; const usersByCity = keyBy(users, 'address.city'); console.log(usersByCity); // 输出: // { // 'New York': { id: '3', name: 'Charlie', address: { city: 'New York' } }, // 'Los Angeles': { id: '2', name: 'Bob', address: { city: 'Los Angeles' } } // }
评论（已关闭）
评论已关闭