Есть ли лучший способ заменить мой двойной метод forEach?

Question

У меня есть два списка, как показано ниже:

var a = ["a", "b"]

var b = [{name:"a1", belong_type:"a" }, {name:"a2", belong_type:"a" }, {name:"b1", belong_type:"b" },]

Я хочу поставить их так:

var data = {}


a.forEach(a_item => {
  data[a_item] = []
  b.forEach(b_item => {
    if (a_item === b_item.belong_type){
      data[a_item].push(b_item)
    }
  })
})

console.log(data)

результат:

{ a: 
   [ { name: 'a1', belong_task_type: 'a' },
     { name: 'a2', belong_task_type: 'a' } ],
  b: [ { name: 'b1', belong_task_type: 'b' } ] }

Я думаю, что мой метод использует два forEach, я не знаю, есть ли лучший способ реализовать результат, кто может сказать мне, если есть лучший способ?

Answer 1

Вы можете использовать метод reduce в массиве a и использовать внутри него метод filter в массиве b для возврата объектов, где belong_type равен текущему элементу в Reduce.

var a = ["a", "b"]
var b = [{name:"a1", belong_type:"a" }, {name:"a2", belong_type:"a" }, {name:"b1", belong_type:"b" }]

const result = a.reduce((r, e) => {
  r[e] = b.filter(({belong_type}) => belong_type == e)
  return r;
}, {})

console.log(result)

Вы также можете использовать Object.assign метод внутри Reduce, чтобы записать его как однострочный.

var a = ["a", "b"]
var b = [{name:"a1", belong_type:"a" }, {name:"a2", belong_type:"a" }, {name:"b1", belong_type:"b" }]

const result = a.reduce((r, e) => Object.assign(r, {[e]: b.filter(({belong_type}) => belong_type == e)}), {})
console.log(result)